Wytrenuj swoje pierwsze drzewo klasyfikacyjne
W tym ćwiczeniu będziesz pracować z Wisconsin Breast Cancer Dataset z repozytorium UCI. Twoim zadaniem będzie przewidzenie, czy guz jest złośliwy, czy łagodny, na podstawie dwóch cech: średniego promienia guza (radius_mean) oraz średniej liczby wklęsłych punktów (concave points_mean).
Zbiór danych jest już wczytany do twojego środowiska i podzielony na 80% danych treningowych i 20% testowych. Macierze cech są przypisane do zmiennych X_train i X_test, natomiast tablice etykiet – do y_train i y_test, gdzie klasa 1 odpowiada guzowi złośliwemu, a klasa 0 – guzowi łagodnemu. Aby zapewnić powtarzalność wyników, zdefiniowano również zmienną SEED o wartości 1.
To ćwiczenie jest częścią kursu
Uczenie maszynowe z modelami drzewiastymi w Pythonie
Instrukcje do ćwiczenia
Zaimportuj
DecisionTreeClassifierz modułusklearn.tree.Utwórz instancję
DecisionTreeClassifiero nazwiedtz maksymalną głębokością równą 6.Dopasuj
dtdo zbioru treningowego.Przewidź etykiety zbioru testowego i przypisz wynik do zmiennej
y_pred.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Import DecisionTreeClassifier from sklearn.tree
from ____.____ import ____
# Instantiate a DecisionTreeClassifier 'dt' with a maximum depth of 6
dt = ____(____=____, random_state=SEED)
# Fit dt to the training set
____.____(____, ____)
# Predict test set labels
y_pred = ____.____(____)
print(y_pred[0:5])