Drop a star to support Aim β | Join Aim discord community |
aimlflow
Run beautiful UI on top of your MLflow logs and get powerful run comparison features. Aim-powered supercharged UI for MLFlow logs
About β’ Getting Started β’ Why use aimlflow? β’ Use Cases β’ Blog
βΉοΈ About
aimlflow helps to explore various types of metadata tracked during the training with MLFLow, including:
- hyper-parameters
- metrics
- images
- audio
- text
More about Aim: https://github.com/aimhubio/aim
More about MLFLow: https://github.com/mlflow/mlflow
π Getting Started
Follow the steps below to set up aimlflow.
- Install aimlflow on your training environment:
pip install aim-mlflow
- Run live time convertor to sync MLFlow logs with Aim:
aimlflow sync --mlflow-tracking-uri={mlflow_uri} --aim-repo={aim_repo_path}
- Run the Aim UI:
aim up --repo={aim_repo_path}
π¦ Why use aimlflow?
- Powerful pythonic search to select the runs you want to analyze.
- Group metrics by hyperparameters to analyze hyperparametersβ influence on run performance.
- Select multiple metrics and analyze them side by side.
- Aggregate metrics by std.dev, std.err, conf.interval.
- Align x axis by any other metric.
- Scatter plots to learn correlations and trends.
- High dimensional data visualization via parallel coordinate plot.
π¬ Use Cases
π Read the article: Exploring MLflow experiments with a powerful UI
π Read the article: How to integrate aimlflow with your remote MLflow
π Read the article: Aim and MLflow β Choosing Experiment Tracker for Zero-Shot Cross-Lingual Transfer