Kom igångKom igång gratis

Träna AdaBoost-klassificeraren

Nu när du har instansierat AdaBoost-klassificeraren ada är det dags att träna den. Du ska också beräkna sannolikheterna för att tillhöra den positiva klassen i testmängden. Så här gör du:

När klassificeraren ada är tränad, anropa metoden .predict_proba() med X_test som argument och extrahera sannolikheterna genom att indexera alla värden i den andra kolumnen:

ada.predict_proba(X_test)[:,1]

Indian Liver-datamängden är förberedd och uppdelad i 80% träningsdata och 20% testdata. Funktionsmatriserna X_train och X_test, samt etikettvektorerna y_train och y_test, finns tillgängliga i din arbetsyta. Dessutom har den instansierade modellen ada från föregående övning laddats in.

Den här övningen är en del av kursen

Maskininlärning med trädbaserade modeller i Python

Visa kurs

Övningsinstruktioner

  • Anpassa ada till träningsdatan.

  • Beräkna sannolikheterna för att tillhöra den positiva klassen i testmängden.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Fit ada to the training set
____

# Compute the probabilities of obtaining the positive class
y_pred_proba = ____.____(____)[____]
Redigera och kör kod