Checking the accuracy of prediction
It’s now time to check how well your trained model can make predictions! Let’s use your testing set to check the accuracy of your Decision Tree model, with the score() method.
This exercise is part of the course
HR Analytics: Predicting Employee Churn in Python
Exercise instructions
- Apply the decision tree
modelto fit thefeaturesto thetargetin the training set. - Check the accuracy
score()of the prediction for the training set. - Check the accuracy
score()of the prediction for the test set.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Apply a decision tree model to fit features to the target in the training set
model.fit(____,____)
# Check the accuracy score of the prediction for the training set
model.____(features_train,____)*100
# Check the accuracy score of the prediction for the test set
model.____(____,____)*100