脚本宝典收集整理的这篇文章主要介绍了scikit-learn学习基础知识二,脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。
本文我们学习scikit-learn中的KNeighborRegressor函数来实现KNN回归进行分类的案例。
"""
K-NN
KNN的一个案例。
KNN一个预测的案例以及进行打分的内容。
"""
import numpy as np
from sklearn.neighbors import KNeighborsRegressor
from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score
X_train = np.array([[158, 1], [170, 1], [183, 1], [191, 1], [155, 0], [163, 0], [180, 0], [158, 0], [170, 0]])
y_train = [64, 86, 84, 80, 49, 59, 67, 54, 67]
X_test = np.array([[168, 1], [180, 1], [160, 0], [169, 0]])
y_test = [65, 96, 52, 67]
K = 3
clf = KNeighborsRegressor(n_neighbors=K)
clf.fit(X_train, y_train)
# 训练模型。
predictions = clf.predict(X_test)
# 进行预测。
print(predictions)
print(r2_score(y_test, predictions))
# 打分。
print(mean_absolute_error(y_test, predictions))
print(mean_squared_error(y_test, predictions))
# 误差。
"""
运行结果。
[70.66666667 79. 59. 70.66666667]
0.6290565226735438
8.333333333333336
95.8888888888889
"""
[70.66666667 79. 59. 70.66666667]
0.6290565226735438
8.333333333333336
95.8888888888889
以上就是使用KNeighborRegressor函数来实现KNN邻近算法的阿里,谢谢大家的阅读与支持啦。
以上是脚本宝典为你收集整理的scikit-learn学习基础知识二全部内容,希望文章能够帮你解决scikit-learn学习基础知识二所遇到的问题。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。