knn_scikit.py 文件源码

def knn_regression(K, training_data, labels, test_data, weights='distance'):
    knn = neighbors.KNeighborsRegressor(K, weights=weights)
    output = knn.fit(training_data, labels).predict(test_data)
    return output