Использование энтропии в качестве критерия
В этом упражнении вы обучите дерево классификации на наборе данных Wisconsin Breast Cancer, используя энтропию в качестве критерия информативности. Для обучения применяются все 30 признаков набора данных, который разделён на 80% обучающую и 20% тестовую выборки.
X_train, а также массив меток y_train доступны в вашем рабочем пространстве.
Это упражнение является частью курса
Машинное обучение на основе древовидных моделей в Python
Инструкции к упражнению
Импортируйте
DecisionTreeClassifierизsklearn.tree.Создайте экземпляр
DecisionTreeClassifierс именемdt_entropyи максимальной глубиной 8.Задайте критерий информативности
'entropy'.Обучите
dt_entropyна обучающей выборке.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Import DecisionTreeClassifier from sklearn.tree
from ____.____ import ____
# Instantiate dt_entropy, set 'entropy' as the information criterion
dt_entropy = ____(____=____, ____='____', random_state=1)
# Fit dt_entropy to the training set
____.____(____, ____)