Menghitung metrik lainnya
Selain akurasi, mari hitung juga skor F1 dari model baru ini agar mendapatkan gambaran yang lebih baik tentang kinerja model.
Pembagian train-test 70-30 sudah disiapkan untuk Anda, dan semua modul yang diperlukan sudah diimpor.
Latihan ini adalah bagian dari kursus
Analitik Pemasaran: Memprediksi Churn Pelanggan di Python
Petunjuk latihan
- Prediksi label himpunan uji.
- Cetak skor F1.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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(____)