DiscordJoin Community

AimStack

 /

aimlflow

0.1.1

pip install aim-mlflow

aimlflow

Aim-powered supercharged UI for MLFlow logs

Run beautiful UI on top of your MLflow logs and get powerful run comparison features.


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.

  1. Install aimlflow on your training environment:
pip install aim-mlflow
  1. Run live time convertor to sync MLFlow logs with Aim:
aimlflow sync --mlflow-tracking-uri={mlflow_uri} --aim-repo={aim_repo_path}
  1. Run the Aim UI:
aim up --repo={aim_repo_path}

Why use aimlflow?

  1. Powerful pythonic search to select the runs you want to analyze.

image

  1. Group metrics by hyperparameters to analyze hyperparameters’ influence on run performance.

image

  1. Select multiple metrics and analyze them side by side.

image

  1. Aggregate metrics by std.dev, std.err, conf.interval.

image

  1. Align x axis by any other metric.

image

  1. Scatter plots to learn correlations and trends.

image

  1. High dimensional data visualization via parallel coordinate plot.

image