开始使用免费开始使用

KNN 分类

在本练习中,您将探索 Large Movie Review Dataset 的一个子集。 变量 X_trainX_testy_trainy_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)
编辑并运行代码