1. Learn
    2. /
  2. Cursuri
    3. /
  3. Machine Learning cu modele bazate pe arbori în Python
Connected

exercițiu

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.

Instrucțiuni

100 XP

  • Antrenează ada pe setul de antrenament.

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