Training two-feature classification models with scikit-learn

We'll continue working on the employee retention problem that we introduced in the first topic. We previously prepared a dataset for training a classification model, in which we predicted whether an employee has left or not. Now, we'll take that data and use it to train classification models:

  1. If you have not already done so, start the NotebookApp and open the chapter-2-workbook.ipynb file. Scroll down to Topic Training classification models. Run the first couple of cells to set the default figure size and load the processed data that we previously saved to a CSV file.

For this example, we'll be training classification models on two continuous features:

 satisfaction_level and ...

Get Applied Deep Learning with Python now with O’Reilly online learning.

O’Reilly members experience live online training, plus books, videos, and digital content from 200+ publishers.