ÎncepețiÎncepe gratuit

Antrenează clasificatorul AdaBoost

Acum că ai instanțiat clasificatorul AdaBoost ada, este momentul să îl antrenezi. Vei calcula și probabilitățile de a obține clasa pozitivă în setul de testare. Iată cum poți face asta:

Odată ce clasificatorul ada este antrenat, apelează metoda .predict_proba() pasând X_test ca parametru și extrage probabilitățile prin selectarea tuturor valorilor din a doua coloană, astfel:

ada.predict_proba(X_test)[:,1]

Setul de date Indian Liver a fost procesat în prealabil și împărțit în 80% date de antrenament și 20% date de testare. Matricele de caracteristici X_train și X_test, precum și vectorii de etichete y_train și y_test sunt disponibili în spațiul tău de lucru. În plus, modelul instanțiat ada din exercițiul anterior a fost deja încărcat.

Acest exercițiu face parte din cursul

Machine Learning cu modele bazate pe arbori în Python

Vezi cursul

Instrucțiuni pentru exercițiu

  • Antrenează ada pe setul de antrenament.

  • Calculează probabilitățile de a obține clasa pozitivă în setul de testare.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Fit ada to the training set
____

# Compute the probabilities of obtaining the positive class
y_pred_proba = ____.____(____)[____]
Editează și rulează codul