tests.py 文件源码

def test_surf_pipeline_cont_endpoint():
    """Ensure that SURF works in a sklearn pipeline with continuous endpoint data"""
    np.random.seed(240932)

    clf = make_pipeline(SURF(n_features_to_select=2, n_jobs=-1),
                        RandomForestRegressor(n_estimators=100, n_jobs=-1))

    assert abs(np.mean(cross_val_score(clf, features_cont_endpoint, labels_cont_endpoint, cv=3))) < 0.5