sklearn - KNeighborsClassifier - ValueError: Unknown label type: 'continuous'

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 回歸器https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsRegressor.html