1. Learn
    2. /
  2. Курси
    3. /
  3. Передобробка для машинного навчання в Python
Connected

вправа

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

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

Інструкції

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