Python Tracker.end_run Examples

Example #1

File: main.py Project: katsugeneration/hydra-mlflow-project-template

def main(config: DictConfig) -> None:
    cwd = pathlib.Path(hydra.utils.get_original_cwd())
    tracker = Tracker(config.experiment, tracking_uri=str(cwd.joinpath('mlruns')))
    tracker.start_run()
    tracker.log_param('runner', config.runner.runner)
    tracker.log_params(config.runner.parameter)

    parameter = container.store[config.runner.runner](**config.runner.parameter)
    runner = container.runner[config.runner.runner](parameter)
    runner.run()
    tracker.log_artifacts(runner.artifacts)
    tracker.log_metrics(runner.metrics)

    tracker.end_run()