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.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
Importez
DecisionTreeClassifierdepuissklearn.tree.Instanciez un
DecisionTreeClassifierdt_entropyavec une profondeur maximale de 8.Définissez le critère d'information à
'entropy'.Entraînez (
fit)dt_entropysur l'ensemble d'entraînement.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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
____.____(____, ____)