mlflow / mlflow-export-import

Apache License 2.0
139 stars 87 forks source link

Implement new click options for experiments #199

Open ptelho-brex opened 2 months ago

ptelho-brex commented 2 months ago

Added options for selective experiment import and model artifact download

We encountered a use case where users didn't want to import each model artifact from each experiment. They also wanted to import only a subset of runs under each experiment, which couldn't easily be done by terminal command.

To address this, I implemented two different click options:

--max-runs [INTEGER]: Allows users to choose the number of runs from each experiment to import. --import-model-artifacts [BOOLEAN]: Gives users the option to download or not download the model artifacts for each experiment.

Example of usage: export-experiments --experiments all --output-dir out --use-threads True --max-runs 10 --import-model-artifacts False