1. Apprendre
    2. /
  2. Cours
    3. /
  3. Machine Learning avec des modèles à base d'arbres en Python
Connected

Exercice

Utiliser l'entropie comme critère

Dans cet exercice, vous allez entraîner un arbre de classification sur l'ensemble de données Wisconsin Breast Cancer en utilisant l'entropie comme critère d'information. Vous utiliserez les 30 variables du jeu de données, séparé en 80 % pour l'entraînement et 20 % pour le test.

X_train ainsi que le tableau d'étiquettes y_train sont disponibles dans votre environnement de travail.

Instructions

100 XP

  • Importez DecisionTreeClassifier depuis sklearn.tree.

  • Instanciez un DecisionTreeClassifier dt_entropy avec une profondeur maximale de 8.

  • Définissez le critère d'information à 'entropy'.

  • Entraînez (fit) dt_entropy sur l'ensemble d'entraînement.