LoslegenKostenlos loslegen

Den AdaBoost-Klassifikator trainieren

Jetzt, da du den AdaBoost-Klassifikator ada instanziiert hast, ist es Zeit, ihn zu trainieren. Außerdem sagst du die Wahrscheinlichkeiten für die positive Klasse im Testset voraus. Das geht so:

Sobald der Klassifikator ada trainiert ist, rufe die Methode .predict_proba() auf, übergib X_test als Parameter und extrahiere die Wahrscheinlichkeiten, indem du alle Werte in der zweiten Spalte wie folgt slicest:

ada.predict_proba(X_test)[:,1]

Der Indian-Liver-Datensatz wurde für dich aufbereitet und in 80 % Training und 20 % Test aufgeteilt. Die Merkmalmatrizen X_train und X_test sowie die Label-Arrays y_train und y_test stehen in deinem Workspace bereit. Zusätzlich haben wir das instanziierte Modell ada aus der vorherigen Übung geladen.

Diese Übung ist Teil des Kurses

Maschinelles Lernen mit baumbasierten Modellen in Python

Kurs anzeigen

Anleitung zur Übung

  • Fitte ada auf den Trainingssatz.

  • Bestimme die Wahrscheinlichkeiten für die positive Klasse im Testset.

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# Fit ada to the training set
____

# Compute the probabilities of obtaining the positive class
y_pred_proba = ____.____(____)[____]
Code bearbeiten und ausführen