---恢复内容开始---
#以iris数据集为例,利用knn算法做预测 #加载数据集 from sklearn import datasets iris = datasets.load_iris() X = iris.data y = iris.target print(X.shape) print(y.shape) from sklearn.neighbors import KNeighborsClassifier #创建实例 knn = KNeighborsClassifier(n_neighbors=1) #模型训练 knn.fit(X, y) #分类预测 x_text = [[1, 2, 3, 4], [3, 2, 2, 1]] res = knn.predict(x_text) print(res)
---恢复内容结束---