Beräkna andra mätvärden
Utöver noggrannhet ska vi också beräkna F1-poängen för den nya modellen, för att få en bättre bild av modellens prestanda.
En 70/30-uppdelning i tränings- och testdata har redan gjorts åt dig, och alla nödvändiga moduler har importerats.
Den här övningen är en del av kursen
Marknadsföringsanalys: Prediktera kundbortfall i Python
Övningsinstruktioner
- Förutsäg etiketterna för testdatan.
- Skriv ut F1-poängen.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Import f1_score
from sklearn.metrics import f1_score
# Instantiate the classifier
clf = RandomForestClassifier()
# Fit to the data
clf.fit(X_train, y_train)
# Predict the labels of the test set
y_pred = ____
# Print the F1 score
print(____)