1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Preprocessing pro Machine Learning v Pythonu
Connected

cvičení

Trénování Naive Bayes s výběrem příznaků

Teď znovu spustíš model textové klasifikace Naive Bayes, který jsi použil/a na konci 3. kapitoly, tentokrát s výběrem příznaků z předchozího cvičení: sloupci title a category_desc z datasetu volunteer.

Pokyny

100 XP
  • Použij train_test_split() na textový vektor filtered_text, štítky y (tedy štítky category_desc) a předej sadu y parametru stratify, protože máme nerovnoměrné rozložení tříd.
  • Natrénuj model Naive Bayes nb na X_train a y_train.
  • Vypočítej přesnost modelu nb na testovací sadě.