1. Nauka
  2. /
  3. Kursy
  4. /
  5. Preprocessing w uczeniu maszynowym w Pythonie

Connected

ćwiczenie

Trenowanie Naive Bayes z selekcją cech

Teraz ponownie uruchomisz model klasyfikacji tekstu Naive Bayes, który uruchomiłeś na końcu rozdziału 3, korzystając z wyborów dokonanych w poprzednim ćwiczeniu: kolumn title i category_desc ze zbioru danych volunteer.

Instrukcje

100 XP
  • Użyj train_test_split() na wektorze tekstowym filtered_text, etykietach y (czyli etykietach category_desc) i przekaż zbiór y do parametru stratify, ponieważ mamy nierównomierny rozkład klas.
  • Dopasuj model Naive Bayes nb do X_train i y_train.
  • Oblicz dokładność modelu nb na zbiorze testowym.