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

Энтропия против индекса Джини

В этом упражнении вы сравните точность dt_entropy на тестовой выборке с точностью другого дерева — dt_gini. Дерево dt_gini было обучено на том же наборе данных с теми же параметрами, за исключением критерия информативности: для него использовался индекс Джини, заданный с помощью ключевого слова 'gini'.

В вашем рабочем пространстве доступны X_test, y_test, dt_entropy, а также accuracy_gini — точность модели dt_gini на тестовой выборке.

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

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

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

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

  • Импортируйте accuracy_score из sklearn.metrics.
  • Предскажите метки тестовой выборки с помощью dt_entropy и сохраните результат в переменную y_pred.
  • Оцените точность dt_entropy на тестовой выборке и сохраните результат в переменную accuracy_entropy.
  • Изучите значения accuracy_entropy и accuracy_gini.

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

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

# Import accuracy_score from sklearn.metrics
from ____.____ import ____

# Use dt_entropy to predict test set labels
____= ____.____(____)

# Evaluate accuracy_entropy
accuracy_entropy = ____(____, ____)

# Print accuracy_entropy
print(f'Accuracy achieved by using entropy: {accuracy_entropy:.3f}')

# Print accuracy_gini
print(f'Accuracy achieved by using the gini index: {accuracy_gini:.3f}')
Редактировать и запускать код