НачатьНачать бесплатно

Оценка дерева классификации

Теперь, когда вы обучили своё первое дерево классификации, пришло время оценить его качество на тестовой выборке. Для этого используйте метрику точности — долю правильных предсказаний на тестовой выборке.

Обученная модель dt из предыдущего упражнения загружена в вашем рабочем пространстве вместе с матрицей признаков тестовой выборки X_test и массивом меток y_test.

Это упражнение является частью курса

Машинное обучение на основе древовидных моделей в Python

Посмотреть курс

Инструкции к упражнению

  • Импортируйте функцию accuracy_score из sklearn.metrics.

  • Предскажите метки тестовой выборки и сохраните полученный массив в переменную y_pred.

  • Вычислите точность модели dt на тестовой выборке с помощью accuracy_score() и сохраните результат в переменную acc.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Import accuracy_score
from ____.____ import ____

# Predict test set labels
y_pred = ____.____(____)

# Compute test set accuracy  
acc = ____(____, ____)
print("Test set accuracy: {:.2f}".format(acc))
Редактировать и запускать код