Python SKLearnWrapper.set_hyperparams示例

示例#1

def test_sklearn_wrapper_update_hyperparams():
    p = SKLearnWrapper(PCA())
    p.set_hyperparams(
        HyperparameterSamples({
            'n_components': 2,
            'svd_solver': 'full'
        }))
    p.update_hyperparams(HyperparameterSamples({'n_components': 4}))

    assert p.wrapped_sklearn_predictor.n_components == 4
    assert p.wrapped_sklearn_predictor.svd_solver == 'full'

示例#2

def test_sklearn_wrapper_set_hyperparams():
    p = SKLearnWrapper(PCA())
    p.set_hyperparams(HyperparameterSamples({'n_components': 2}))

    assert p.wrapped_sklearn_predictor.n_components == 2