本文介绍了如何在Python的scikit-learn库中使用KNeighborsClassifier进行分类，以及KNeighborsRegressor进行回归预测。通过实例展示了如何实例化、训练模型并进行预测，如对数据点(3,11,10)进行回归预测。
#todo 导包
from sklearn.neighbors import KNeighborsClassifier
#todo 导数据
x = [[0],[1],[2],[3]]
y = [0,0,1,1]
#todo 实例化对象
model = KNeighborsClassifier(n_neighbors=2)
#todo 训练
model.fit(x,y)
#todo 预测
mypre = model.predict([[1.5]])
print(f'预测值:{mypre}')sklearn.neighbors.KNeighborsClassifier — scikit-learn 1.4.2 documentation
from sklearn.neighbors import KNeighborsRegressor
def dm02_knnapi_回归():
estimator = KNeighborsRegressor(n_neighbors=3)
X = [[0, 0, 1],
[1, 1, 0],
[3, 10, 10],
[4, 11, 12]]
y = [0.1, 0.2, 0.3, 0.4]
estimator.fit(X, y)
myret = estimator.predict([[3, 11, 10]])
print('myret-->', myret)
dm02_knnapi_回归()sklearn.neighbors.KNeighborsRegressor — scikit-learn 1.4.2 documentation
1796
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
