# !/usr/bin/env python3
from sklearn import datasets
from sklearn.neighbors import KNeighborsClassifier
import matplotlib.pyplot as plt
import numpy as np
iris = datasets.load_iris()
x = iris.data
y = iris.target
h = y.reshape(-1,1)
'''将特征值'''
data = np.concatenate((x,h),axis=1)
'''随机打乱数据'''
np.random.shuffle(data)
'''获取训练集和测试集'''
train_data = data[:int(len(data)*0.8)]
test_data = data[int(len(data)*0.8):]
'''训练集'''
X_train = train_data[:,0:4]
Y_train = train_data[:,4]
'''测试集'''
X_test = test_data[:,0:4]
Y_test = test_data[:,4]
kNN_classifier = KNeighborsClassifier(n_neighbors=6)
kNN_classifier.fit(X_train,Y_train)
'''预测'''
Y_predict = kNN_classifier.predict(X_test)
'''测试结果 np.sum 会统计元素中所有为 true的为1 false的为0 并且相加'''
print(np.sum(Y_predict == Y_test)/len(Y_test))
使用KNN近邻算法学习鸢尾花数据
最新推荐文章于 2024-10-19 16:02:10 发布