1. Learn
    2. /
  2. Courses
    3. /
  3. Preprocessing for Machine Learning in Python

Exercise

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.

Instructions

100 XP
  • Use train_test_split() on the filtered_text text vector, the y labels (which is the category_desc labels), and pass the y set to the stratify parameter, since we have an uneven class distribution.
  • Fit the nb Naive Bayes model to X_train and y_train.
  • Calculate the test set accuracy of nb.