机器学习sklearn入门

---恢复内容开始---

#以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)

---恢复内容结束---