knn算法

import numpy as np
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier

def knn_sklearn():
#加载iris数据
li = load_iris()
#分割测试集和训练集,测试集占整个数据集的比例是0.25
x_train,x_test,y_train,y_test = train_test_split(li.data,li.target,test_size=0.25)
#创建KNN分类器,使用最少5个邻居作为类别判断标准
knn = KNeighborsClassifier(n_neighbors=5)
#训练数据
knn.fit(x_train,y_train)
#预测[6.3,3,5.2,2.3]所属各个类别的概率
print(knn.predict_proba(np.array([[6.3,3,5.2,2.3]])))

knn_sklearn()