knn分类

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

# 载入数据
iris = datasets.load_iris()
iris_X = iris.data
iris_y = iris.target

train_d, test_d, train_tar, test_tar = train_test_split(iris_X, iris_y, test_size=0.2, random_state=5)
# 随机划分测试集和训练集

knn = KNeighborsClassifier(weights='distance')
knn.fit(train_d, train_tar)

# 用训练好的函数预测

print(knn.predict(train_d))
print(train_tar)
print(knn.predict(test_d))
print(test_tar)
print(knn.predict_proba(test_d))

# 返回每个类的预测值