Esempi in Python per StackingRegressor.get_params

Esempio n. 1

def test_get_params():
    lr = LinearRegression()
    svr_rbf = SVR(kernel='rbf')
    ridge = Ridge(random_state=1)
    stregr = StackingRegressor(regressors=[ridge, lr], meta_regressor=svr_rbf)

    got = sorted(list({s.split('__')[0] for s in stregr.get_params().keys()}))
    expect = [
        'linearregression', 'meta-svr', 'meta_regressor', 'regressors',
        'ridge', 'store_train_meta_features', 'verbose'
    ]
    assert got == expect, got

Esempio n. 2

File: test_stacking_regression.py Progetto: venkatesh-1729/mlxtend

def test_get_params():
    lr = LinearRegression()
    svr_rbf = SVR(kernel='rbf')
    ridge = Ridge(random_state=1)
    stregr = StackingRegressor(regressors=[ridge, lr],
                               meta_regressor=svr_rbf)

    got = sorted(list({s.split('__')[0] for s in stregr.get_params().keys()}))
    expect = ['linearregression',
              'meta-svr',
              'meta_regressor',
              'regressors',
              'ridge',
              'store_train_meta_features',
              'verbose']
    assert got == expect, got