Canalización para predecir el género de las canciones II
Una vez configurados los pasos de la canalización en el ejercicio anterior, ahora lo utilizarás en el conjunto de datos music_df para clasificar el género de las canciones. Lo que hace que las canalizaciones sean tan increíblemente útiles es la sencilla interfaz que proporcionan.
Se ha precargado X_train, X_test, y_train e y_test, y se ha importado confusion_matrix de sklearn.metrics.
Este ejercicio forma parte del curso
Aprendizaje supervisado con scikit-learn
Instrucciones del ejercicio
- Crea una canalización utilizando los pasos que definiste anteriormente.
- Ajusta la canalización a los datos de entrenamiento.
- Haz predicciones sobre el conjunto de pruebas.
- Calcula e imprime la matriz de confusión.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
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(____(____, ____))