首页 > 解决方案 > 在 K-NN 中,如何显示用于做出正确预测的 k 个案例集

问题描述

(k=10)我训练并测试了一个在 python 中使用 K-NN 预测销售额的模型。

knn = KNeighborsRegressor(n_neighbors=10, algorithm="brute") 
knn.fit(X_train, y_train)
knn_pred = knn.predict(X_test) 
test_data['predictied'] = knn_pred

我想展示由 K-NN 选择的 10 个案例集,以对每一行进行预测。我如何显示或检索这些k案例?

标签: pythonmachine-learningknn

解决方案


文档

kneighbors(self, X=None, n_neighbors=None, return_distance=True)

查找一个点的 K 邻域。返回每个点的邻居的索引和距离。

此方法将返回用于计算值的点的距离和索引


推荐阅读