Конвеєр для передбачення жанру пісні: 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(____(____, ____))