Entropie als Kriterium verwenden
In dieser Übung trainierst du einen Klassifikationsbaum auf dem Wisconsin Breast Cancer Datensatz und verwendest Entropie als Informationskriterium. Du nutzt alle 30 Features des Datensatzes, der in 80 % Training und 20 % Test aufgeteilt ist.
X_train sowie das Label-Array y_train stehen dir in deiner Arbeitsumgebung zur Verfügung.
Diese Übung ist Teil des Kurses
Maschinelles Lernen mit baumbasierten Modellen in Python
Anleitung zur Übung
Importiere
DecisionTreeClassifieraussklearn.tree.Instanziiere einen
DecisionTreeClassifierdt_entropymit einer maximalen Tiefe von 8.Setze das Informationskriterium auf
'entropy'.Fitte
dt_entropyauf dem Trainingssatz.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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
____.____(____, ____)