2021-03-16 深度学习(二)使用scikit-learn 中的kNN

本文介绍了如何在深度学习中应用scikit-learn库进行KNN算法的训练和预测。通过调用fit方法进行模型拟合,然后利用predict方法进行数据预测。
摘要由CSDN通过智能技术生成

把X_train,y_train送入机器学习算法,机器学习算法训练模型的过程称为拟合(fit),也就是说我们的算法要训练模型,这个模型要能够拟合我们的数据集,把输入样例输入模型得出结果的过程称为预测(predict).

使用scikit-learn 中的CNN:

from sklearn.neighbors import KNeighborsClassifier
#在sklearn库中,KNeighborsClassifier是实现K近邻算法的一个类,一般都使用欧式距离进行测量。
kNN_classifier = KNeighborsClassifier(n_neighbors=6)#创建实例
kNN_classifier.fit(X_train,y_train) #拟合我们的训练集

输出KNeighborsClassifier(n_neighbors=6)

X_predict=x.reshape(1,-1)#传1代表第一个维度,-1让numpy自动决定到哪截止
X_predict

输出array([[8.09360732, 3.36573151]])

kNN_classifier.predict(X_predict)

输出array([1])
​

封装:

import nump
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值