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

Вправа

Моделювання набору даних про НЛО, частина 2

Нарешті ви побудуєте модель, використовуючи створений нами текстовий вектор desc_tfidf, а також список filtered_words, щоб сформувати відфільтрований текстовий вектор. Перевірмо, чи зможете ви передбачити type спостереження на основі тексту. Для цього ви використаєте наївний Байєсівський класифікатор.

Інструкції

100 XP
  • Відфільтруйте вектор desc_tfidf, передавши до індексу список filtered_words.
  • Розбийте ознаки filtered_text і y, забезпечивши однаковий розподіл класів у тренувальній та тестовій вибірках; використайте random_state зі значенням 42.
  • Використайте .fit() моделі nb, щоб натренувати X_train і y_train.
  • Виведіть .score() моделі nb на вибірках X_test та y_test.