LoslegenKostenlos loslegen

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

Kurs anzeigen

Anleitung zur Übung

  • Importiere DecisionTreeClassifier aus sklearn.tree.

  • Instanziiere einen DecisionTreeClassifier dt_entropy mit einer maximalen Tiefe von 8.

  • Setze das Informationskriterium auf 'entropy'.

  • Fitte dt_entropy auf 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
____.____(____, ____)
Code bearbeiten und ausführen