В K-NN Как показать набор из k случаев, который был использован для правильного прогноза - PullRequest
0 голосов
/ 08 января 2020

Я обучил и протестировал модель, которая прогнозирует продажи, используя K-NN с (k=10) в python.

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

Я хочу показать набор из 10 случаев, который был выбран K- NN, чтобы сделать прогноз для каждой строки. Как я могу показать или получить это k случаев?

1 Ответ

1 голос
/ 08 января 2020

Из документации

соседей (self, X = None, n_neighbors = None, return_distance = True)

Находит K-соседей точки , Возвращает индексы и расстояния до соседей каждой точки.

, этот метод возвращает расстояния и индекс точек, использованных для вычисления значения

...