Конвейер для определения жанра песни: II
На предыдущем шаге вы настроили этапы конвейера. Теперь применим его к набору данных music_df для классификации жанров песен. Конвейеры особенно удобны благодаря простому и единообразному интерфейсу.
Переменные X_train, X_test, y_train и y_test уже загружены, а confusion_matrix импортирована из sklearn.metrics.
Это упражнение является частью курса
Обучение с учителем с помощью scikit-learn
Инструкции к упражнению
- Создайте конвейер, используя ранее определённые шаги.
- Обучите конвейер на обучающих данных.
- Сделайте предсказания на тестовой выборке.
- Вычислите матрицу ошибок и выведите её на экран.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
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(____(____, ____))