test.py 文件源码

def test(x_file, y_file, train_list, test_list, best_params):
    X1 = np.loadtxt(x_file, delimiter=",")
    Y1 = np.loadtxt(y_file, delimiter=",")
    train_X, train_Y, test_X, _ = split_train_val(X1, Y1, train_list, test_list)
    # print train_X.shape,test_X.shape
    EXT1 = ExtraTreesRegressor(n_jobs=-1, random_state=1, **best_params)
    EXT1.fit(train_X, train_Y)
    test_Y1 = EXT1.predict(test_X)
    # print EXT1.feature_importances_
    return test_Y1