Learn Python by Building Data Science Applications

Exploring feature engineering

Now that we made a system to fairly compare models with no fear of overfitting, let's think about how we can improve our model. One way would be to create new features that might add more context. One way to go about this is to create features of our own, for example, calculate a proportion of armies on different sides or the absolute difference in the number of soldiers—we can't say in advance which would work better. Let's try it out with the help of the following code:

1. First, we'll create a ratio of soldiers on either side:

data['infantry_ratio'] = data['allies_infantry'] / data['axis_infantry']cols.append('infantry_ratio')

2. Now, we won't do that for tanks, planes, and so on, as the numbers here are very ...