Python Artificial Intelligence Projects for Beginners

male
 or
female
, we will know whether he or she survived, hence the information gain is highest for
the sex column. We do not consider age column best for our first split since we do not know
much about the passengers ages, and is not the best first split because we will know less
about the outcome if all we know is a passenger's age.
After splitting on the 
sex
 column according to the information gain, what we have now
is 
female
 and 
male
 subsets, as seen in the following screenshot:

---

Building Your Own Prediction Models
Chapter 1
[ 13 ]
After the split, we have one internode and one question node, as seen in the previous
screenshot, and two paths that can be taken depending on the answer to the question. Now
we need to find the best attribute again in both of the subsets. The left subset, in which all
passengers are female, does not have a good attribute to split on because many passengers
survived. Hence, the left subset just turns into a leaf node that predicts survival. On the
right-hand side, the 
BHF
 attribute is chosen as the best split, considering the value 
9.5
 years
of age as the split. We gain two more subsets: age greater than