KNN 分类
在本练习中,您将探索 Large Movie Review Dataset 的一个子集。
变量 X_train、X_test、y_train 和 y_test 已加载到环境中。X 变量包含基于影评文本的特征,y 变量包含情感标签:正面为 +1,负面为 -1。
本课程会涉及许多您可能已经遗忘的概念。如果需要快速复习,欢迎下载并随身查看 scikit-learn 速查表。
本练习是课程的一部分
Python 中的线性分类器
练习说明
- 使用默认超参数创建一个 KNN 模型。
- 训练该模型。
- 打印测试样本 0 的预测结果。
交互式实操练习
通过完成这段示例代码来试试这个练习。
from sklearn.neighbors import KNeighborsClassifier
# Create and fit the model
knn = ____
knn.____
# Predict on the test features, print the results
pred = knn.____[0]
print("Prediction for test example 0:", pred)