Zacznij terazZacznij za darmo

Entropia jako kryterium podziału

W tym ćwiczeniu nauczysz drzewo klasyfikacyjne na zbiorze danych Wisconsin Breast Cancer, używając entropii jako kryterium informacyjnego. Skorzystasz ze wszystkich 30 cech w zbiorze danych, który jest podzielony na 80% danych treningowych i 20% testowych.

X_train oraz tablica etykiet y_train są dostępne w twoim środowisku pracy.

To ćwiczenie jest częścią kursu

Uczenie maszynowe z modelami drzewiastymi w Pythonie

Zobacz kurs

Instrukcje do ćwiczenia

  • Zaimportuj DecisionTreeClassifier z sklearn.tree.

  • Utwórz instancję DecisionTreeClassifier o nazwie dt_entropy z maksymalną głębokością równą 8.

  • Ustaw kryterium informacyjne na 'entropy'.

  • Dopasuj dt_entropy do zbioru treningowego.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# 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
____.____(____, ____)
Edytuj i uruchom kod