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

Использование энтропии в качестве критерия

В этом упражнении вы обучите дерево классификации на наборе данных 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
____.____(____, ____)
Редактировать и запускать код