1. Learn
    2. /
  2. 课程
    3. /
  3. Python 中的机器学习预处理
Connected

道练习

使用特征选择训练朴素贝叶斯

现在，您将使用上一个练习中的选择结果，重新运行第 3 章末尾实现过的朴素贝叶斯文本分类模型：volunteer 数据集中的 title 和 category_desc 列。

说明

100 XP
  • 对 filtered_text 文本向量与 y 标签（即 category_desc 标签）使用 train_test_split()，并将 y 传给 stratify 参数，因为类别分布不均衡。
  • 将 nb 朴素贝叶斯模型拟合到 X_train 和 y_train。
  • 计算 nb 的测试集准确率。