1. Learn
    2. /
  2. Courses
    3. /
  3. Kaggle R Tutorial on Machine Learning

Exercise

Passenger Title and survival rate

Was it coincidence that upper-class Rose survived and third-class passenger Jack not? Let's have a look...

You have access to a new train and test set named train_new and test_new. These data sets contain a new column with the name Title (referring to Miss, Mr, etc.). Title is another example of feature engineering: it's a new variable that possibly improves the model.

Instructions

100 XP
  • Finish the command to create a decision tree my_tree_five: make sure to include the Title variable, and to create the tree based on train_new.
  • Visualize my_tree_five with fancyRpartPlot(). Notice that Title appears in one of the nodes.
  • Finish the predict() call to create my_prediction: the function should use my_tree_five and test_new to make predictions.
  • The code that creates a data frame my_solution and writes it to a CSV file is included: these steps make the solution ready for a submission on Kaggle.