/sweep

A minimalist sweep tool.

Primary LanguagePythonMIT LicenseMIT

Sweep: A minimalist sweep tool

Python Version Code Style

Installation | Example Usage

This library allows to simply evaluate all combinations of given hyperparameters.

Installation

pip install --upgrade pip
pip install --upgrade git+https://github.com/Raffaelbdl/sweep

Example Usage

import sweep 

def main(cfg):
    param_A = cfg.pop("param_A", 1)
    param_B = cfg.pop("param_B", 2)

    # do stuff
    return 

if __name__ == "__main__":
    sweep_config = {
        "A": [0, 1, 2],
        "B": [0, 1, 2],
    }
    sweep.run(main, sweep_config)