KNN应用

介绍

用KNN算法来拟合鸢尾花数据，然后给出一个自定义输入，预测鸢尾花的类别

数据集

鸢尾花，150个实例，三个类别，4个特征

PythonCode

# 导入库
from sklearn import neighbors
# 导入数据集
from sklearn import datasets

# 生产KNN实例
knn = neighbors.KNeighborsClassifier()
# 加载数据
iris = datasets.load_iris()

data = iris.get('data')
target = iris.get('target')
target_names = iris.get('target_names')

print(iris.data)
print(target)
print(target_names)

# 拟合数据
knn.fit(iris.data, iris.target)

# 进行预测
predictT = knn.predict([[0.1, 0.2, 0.3, 0.4]])
preName = target_names[predictT]

print(preName)