2021-09-06 分类 - K最近邻算法

最新推荐文章于 2024-11-12 21:59:57 发布

ywq4321

最新推荐文章于 2024-11-12 21:59:57 发布

阅读量132

点赞数 1

分类专栏： Python人工智能文章标签：算法 python 人工智能

本文链接：https://blog.csdn.net/ywq4321/article/details/120143435

版权

Python人工智能专栏收录该内容

2 篇文章 0 订阅

订阅专栏

在这里插入图片描述


#K最近邻算法 - KNN, kNearestNeighbor
import numpy as np
import pandas as pd
from sklearn import neighbors
dic1 = {'id':[0,1,2,3,4,5,6,7,8],'yvwen':[72,66,89,81,86,91,86,89,94],'shuxue':[92,81,80,79,68,79,74,78,61], 'ClassifyResult':[3,3,2,2,1,1,1,1,1]}
df = pd.DataFrame(dic1)
print(df)
train_x = df.iloc[0:7,1:3] #训练集的输入
print(train_x)
train_y = df.iloc[0:7,3] #训练集的输出
print(train_y)
model = neighbors.KNeighborsClassifier()
model.fit(train_x,train_y)
print('###')
print(model)
test_x = df.iloc[7:9,1:3] #测试集的输入
print(test_x)
test_p = model.predict(test_x) #预测值
print(test_p)
test_y = df.iloc[7:9,3].values #测试集的输出
print(test_y)
print(model.score(test_x,test_y)) #正确率