Antrenează primul tău arbore de clasificare
În acest exercițiu vei lucra cu Wisconsin Breast Cancer Dataset din depozitul de machine learning al UCI. Vei prezice dacă o tumoare este malignă sau benignă pe baza a două caracteristici: raza medie a tumorii (radius_mean) și numărul mediu de puncte concave (concave points_mean).
Setul de date este deja încărcat în spațiul tău de lucru și este împărțit în 80% antrenament și 20% testare. Matricele de caracteristici sunt atribuite variabilelor X_train și X_test, iar vectorii de etichete sunt atribuiți variabilelor y_train și y_test, unde clasa 1 corespunde unei tumori maligne, iar clasa 0 corespunde unei tumori benigne. Pentru a obține rezultate reproductibile, am definit și o variabilă numită SEED, cu valoarea 1.
Acest exercițiu face parte din cursul
Machine Learning cu modele bazate pe arbori în Python
Instrucțiuni pentru exercițiu
Importă
DecisionTreeClassifierdinsklearn.tree.Instanțiază un
DecisionTreeClassifiernumitdtcu adâncimea maximă egală cu 6.Antrenează
dtpe setul de antrenament.Prezice etichetele setului de testare și atribuie rezultatul variabilei
y_pred.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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])