offlax.runner#

Classes

OfflaxRunner(algorithm[, search_space, …])

Generic runner that encapsulates search, training and inference.
class offlax.runner.OfflaxRunner(algorithm, search_space=None, enable_wandb=True, wandb_kwargs={}, *args, **kwargs)[source]#

Bases: object

Generic runner that encapsulates search, training and inference.

Parameters

  • algorithm (str) – Offlax Algorithm to run

  • search_space (_type_) – Search space of the algorithm. Defaults to the algorithm’s search space if unspecified

  • enable_wandb (bool) –

  • wandb_kwargs (Dict) –