ПочатиПочніть безкоштовно

Навчання Naive Bayes із вибором ознак

Тепер ви повторно запустите модель класифікації тексту Naive Bayes, яку ви запускали наприкінці Розділу 3, використовуючи вибір із попередньої вправи: стовпці title і category_desc з набору даних volunteer.

Ця вправа є частиною курсу

Передобробка для машинного навчання в Python

Переглянути курс

Інструкції до вправи

  • Застосуйте train_test_split() до текстового вектора filtered_text, міток y (це мітки category_desc) і передайте множину y параметру stratify, адже розподіл класів нерівномірний.
  • Навчіть модель Naive Bayes nb на X_train і y_train.
  • Обчисліть точність моделі nb на тестовій вибірці.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

# Split the dataset according to the class distribution of category_desc
X_train, X_test, y_train, y_test = ____(____.toarray(), ____, stratify=____, random_state=42)

# Fit the model to the training data
nb.____

# Print out the model's accuracy
print(nb.____)
Редагувати та запускати код