1. Learn
    2. /
  2. Курси
    3. /
  3. Машинне навчання з деревоподібними моделями в Python
Connected

вправа

Використання ентропії як критерію

У цій вправі ви навчите класифікаційне дерево на наборі даних Wisconsin Breast Cancer, використовуючи ентропію як інформаційний критерій. Ви зробите це, використавши всі 30 ознак у наборі даних, який поділено на 80% тренування і 20% тест.

X_train, а також масив міток y_train, доступні у вашому робочому середовищі.

Інструкції

100 XP

  • Імпортуйте DecisionTreeClassifier з sklearn.tree.

  • Створіть екземпляр DecisionTreeClassifier dt_entropy із максимальною глибиною 8.

  • Встановіть інформаційний критерій у 'entropy'.

  • Навчіть dt_entropy на тренувальній вибірці.