Предсказание съедобности грибов
Теперь, когда вы изучили данные, пришло время построить первую модель для предсказания съедобности грибов.
Набор данных доступен вам как mushrooms. Поскольку и признаки, и целевая переменная являются категориальными, они уже преобразованы в «фиктивные» бинарные переменные.
Начнём с наивного байесовского классификатора (с использованием GaussianNB из scikit-learn) и оценим, как этот алгоритм справляется с задачей.
Это упражнение является частью курса
Ансамблевые методы в Python
Инструкции к упражнению
- Создайте экземпляр классификатора
GaussianNBс именемclf_nb. - Обучите
clf_nbна обучающих данныхX_trainиy_train. - Вычислите предсказания на тестовой выборке. Они будут использованы для оценки качества модели с помощью метрики accuracy.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Instantiate a Naive Bayes classifier
clf_nb = ____
# Fit the model to the training set
____
# Calculate the predictions on the test set
pred = ____
# Evaluate the performance using the accuracy score
print("Accuracy: {:0.4f}".format(accuracy_score(y_test, pred)))