search

automl / neps

Neural Pipeline Search (NePS): Helps deep learning experts find the best neural pipeline.
https://automl.github.io/neps/
Apache License 2.0
39 stars 11 forks source link

Custom searcher ignoring max_cost_total #75

Open danrgll opened 2 months ago

danrgll commented 2 months ago

When utilizing a custom searcher, the parameter max_cost_total is currently being ignored.

To make clear what scenario I mean, see this example:

def run_pipeline(**config):
    return {"loss": 1, "cost": 3}

search_space = SearchSpace(**pipeline_space)
optimizer = BayesianOptimizer(search_space)
neps.run(run_pipeline, searcher=optimizer, max_cost_total=100)