4.2 KNN regressor.py 文件源码

def test_KNeighborsRegressor(*data):
    '''
    test the KNN regressor
    :param data: train_data, test_data, train_value, test_value
    :return: None
    '''
    X_train,X_test,y_train,y_test=data
    regr=neighbors.KNeighborsRegressor()
    regr.fit(X_train,y_train)
    print("Training Score:{0}".format(regr.score(X_train,y_train)))
    print("Testing Score:{0}".format(regr.score(X_test,y_test)))