scikit-learn学习基础知识二

Posted 2022-02-05 hhh江月

scikit-learn学习基础知识二

文章目录

• • 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邻近算法的阿里，谢谢大家的阅读与支持啦。