1. Learn
    2. /
  2. Курси
    3. /
  3. Prétraitement pour le Machine Learning en Python
Connected

вправа

Entraîner Naive Bayes avec sélection de caractéristiques

Vous allez maintenant relancer le modèle de classification de texte Naive Bayes utilisé à la fin du chapitre 3, en appliquant nos choix de sélection de l'exercice précédent : les colonnes title et category_desc de l'ensemble de données volunteer.

Інструкції

100 XP
  • Utilisez train_test_split() sur le vecteur de texte filtered_text, les étiquettes y (qui correspondent aux étiquettes category_desc), et passez l'ensemble y au paramètre stratify, puisque la distribution des classes est inégale.
  • Ajustez le modèle Naive Bayes nb sur X_train et y_train.
  • Calculez la précision sur l'ensemble de test de nb.