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(____(____, ____))