Evaluate the classification tree
Now that you've fit your first classification tree, it's time to evaluate its performance on the test set. You'll do so using the accuracy metric which corresponds to the fraction of correct predictions made on the test set.
The trained model dt
from the previous exercise is loaded in your workspace along with the test set features matrix X_test
and the array of labels y_test
.
This exercise is part of the course
Machine Learning with Tree-Based Models in Python
Exercise instructions
Import the function
accuracy_score
fromsklearn.metrics
.Predict the test set labels and assign the obtained array to
y_pred
.Evaluate the test set accuracy score of
dt
by callingaccuracy_score()
and assign the value toacc
.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Import accuracy_score
from ____.____ import ____
# Predict test set labels
y_pred = ____.____(____)
# Compute test set accuracy
acc = ____(____, ____)
print("Test set accuracy: {:.2f}".format(acc))