Training Naive Bayes with feature selection
You'll now re-run the Naive Bayes text classification model that you ran at the end of Chapter 3 with our selection choices from the previous exercise: the volunteer dataset's title and category_desc columns.
Cet exercice fait partie du cours
Preprocessing for Machine Learning in Python
Instructions
- Use
train_test_split()on thefiltered_texttext vector, theylabels (which is thecategory_desclabels), and pass theyset to thestratifyparameter, since we have an uneven class distribution. - Fit the
nbNaive Bayes model toX_trainandy_train. - Calculate the test set accuracy of
nb.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Split the dataset according to the class distribution of category_desc
X_train, X_test, y_train, y_test = ____(____.toarray(), ____, stratify=____, random_state=42)
# Fit the model to the training data
nb.____
# Print out the model's accuracy
print(nb.____)