Энтропия против индекса Джини
В этом упражнении вы сравните точность 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}')