ÎncepețiÎncepe gratuit

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

Vezi cursul

Instrucțiuni pentru exercițiu

  • Importă DecisionTreeClassifier din sklearn.tree.

  • Instanțiază un DecisionTreeClassifier numit dt cu adâncimea maximă egală cu 6.

  • Antrenează dt pe 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])
Editează și rulează codul