MCPcopy Index your code
hub / github.com/aimhubio/aimlflow

github.com/aimhubio/aimlflow @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
42 symbols 166 edges 7 files 2 documented · 5%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README
Powered by Aim
Drop a star to support Aim ⭐ Join Aim discord community

aimlflow

Aim-powered supercharged UI for MLFlow logs

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

Platform Support PyPI - Python Version PyPI Package License

AboutGetting StartedWhy 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.

  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

🎬 Use Cases

🎇 Read the article: Exploring MLflow experiments with a powerful UI

image

🔍 Read the article: How to integrate aimlflow with your remote MLflow

image

📊 Read the article: Aim and MLflow — Choosing Experiment Tracker for Zero-Shot Cross-Lingual Transfer

image

More questions?

  1. Read the docs
  2. Open a feature request or report a bug
  3. Join Discord community server

Core symbols most depended-on inside this repo

get
called by 6
aimlflow/utils.py
status
called by 4
setup.py
get_mlflow_experiments
called by 3
aimlflow/utils.py
_process_single_run
called by 2
aimlflow/watcher.py
_process_runs
called by 2
aimlflow/watcher.py
refresh
called by 2
aimlflow/utils.py
get_aim_run
called by 2
aimlflow/utils.py
collect_run_params
called by 2
aimlflow/utils.py

Shape

Method 23
Function 14
Class 5

Languages

Python100%

Modules by API surface

aimlflow/utils.py15 symbols
aimlflow/watcher.py9 symbols
examples/hparam-tuning/tune.py6 symbols
setup.py5 symbols
examples/cross-lingual-transfer/main.py5 symbols
aimlflow/cli.py2 symbols

For agents

$ claude mcp add aimlflow \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact