1. Aprende
    2. /
  2. Cursos
    3. /
  3. Supervised Learning with scikit-learn
Connected

Ejercicio

Pipeline for song genre prediction: I

Now it's time to build a pipeline. It will contain steps to impute missing values using the mean for each feature and build a KNN model for the classification of song genre.

The modified music_df dataset that you created in the previous exercise has been preloaded for you, along with KNeighborsClassifier and train_test_split.

Instrucciones

100 XP
  • Import SimpleImputer and Pipeline.
  • Instantiate an imputer.
  • Instantiate a KNN classifier with three neighbors.
  • Create steps, a list of tuples containing the imputer variable you created, called "imputer", followed by the knn model you created, called "knn".