a03930f6a94bb772222d00b819c3ff824e28ddcc,cesium/tests/test_build_model.py,,test_fit_optimize,#,69

Before Change

@with_setup(teardown=remove_output)
def test_fit_optimize():
    Test hypeparameter optimization
    fset = xr.open_dataset(pjoin(DATA_PATH, "asas_training_subset_featureset.nc"))
    model = build_model.MODELS_TYPE_DICT["RandomForestClassifier"]()
    model_options = {"criterion": "gini", "bootstrap": True}
    model = build_model.MODELS_TYPE_DICT["RandomForestClassifier"]\
            (**model_options)

After Change

@with_setup(teardown=remove_output)
def test_fit_optimize():
    Test hypeparameter optimization
    fset = sample_featureset(10, ["amplitude", "maximum", "minimum", "median"],
                             ["class1", "class2"])
    model_options = {"criterion": "gini", "bootstrap": True}
    model = build_model.MODELS_TYPE_DICT["RandomForestClassifier"]\
            (**model_options)
    feature_df = build_model.rectangularize_featureset(fset)

Instances