sklearn - KNeighborsClassifier - ValueError: Неизвестный тип метки: «непрерывный»

Question

Не знаю, почему этот код не работает?

    training_data_X    = np.array([ [1.2, 6.7, 2.7],  [2.3, 4.6, 2.2],  [0.3, 3.9, 0.8],  [2.1, 1.3, 4.3]  ])
    training_scores_Y  = np.array( [1.4, 9.2, 2.5, 2.2] )
    y_test  = np.array([ [1.2, 6.7, 2.7],  [7.6, 7.2, 0.2] ])

    knn = KNeighborsClassifier(n_neighbors=1)
    knn.fit(training_data_X, training_scores_Y)

    y_pred = knn.predict(y_test)
    print(y_pred)

Я получил эту ошибку:

ValueError: Unknown label type: 'continuous'

Спасибо за вашу помощь.

Answer 1

• Вы использовали классификатор KNN, в то время как ваша проблема в регрессии.

• Вы можете использовать KNN Regressor https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsRegressor.html