Pipeline untuk prediksi genre lagu: II
Setelah menyiapkan langkah-langkah pipeline pada latihan sebelumnya, sekarang Anda akan menggunakannya pada himpunan data music_df untuk mengklasifikasikan genre lagu. Keunggulan besar dari pipeline adalah antarmukanya yang sederhana.
X_train, X_test, y_train, dan y_test telah dimuatkan untuk Anda, dan confusion_matrix telah diimpor dari sklearn.metrics.
Latihan ini adalah bagian dari kursus
Supervised Learning dengan scikit-learn
Petunjuk latihan
- Buat sebuah pipeline menggunakan langkah-langkah yang telah Anda definisikan sebelumnya.
- Fit pipeline pada data latih.
- Lakukan prediksi pada himpunan data uji.
- Hitung dan cetak confusion matrix.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
steps = [("imputer", imp_mean),
("knn", knn)]
# Create the pipeline
pipeline = ____(____)
# Fit the pipeline to the training data
____
# Make predictions on the test set
y_pred = ____
# Print the confusion matrix
print(____(____, ____))