jutil.py 文件源码

def gs_ElasticNet( xM, yV, en_params):

    print(xM.shape, yV.shape)

    clf = linear_model.ElasticNet()
    kf5_c = model_selection.KFold( n_folds=5, shuffle=True)
    kf5 = kf5_c.split( xM) 
    gs = model_selection.GridSearchCV( clf, en_params, scoring = 'r2', cv = kf5, n_jobs = -1)

    gs.fit( xM, yV)

    return gs