我使用Python中的K-NN模型((k=10)
)训练并测试了一个预测销售的模型。
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
个案例?
回答:
来自文档
kneighbors(self, X=None, n_neighbors=None, return_distance=True)
查找一个点的K个邻居。返回每个点的邻居的索引和距离。
此方法将返回用于计算值的点的距离和索引