Jupyter Notebook Example

Setup

!pip install aime-loc[viz]

Interactive Exploration

from aime_loc import LOC

loc = LOC()

# Scan a model
profile = loc.scan("meta-llama/Llama-4-Scout")

# Display inline radar chart
profile.radar_chart()

Comparison Dashboard

# Compare two models
comp = loc.compare("Llama-4-Scout", "DeepSeek-R1")

# Side-by-side radar
comp.side_by_side_radar()

# Delta chart
comp.delta_chart()

# Summary
print(comp.summary())

Export for Paper

# Save figures
profile.export_figure("fig1_radar.pdf", journal="nature", fmt="pdf")
profile.to_csv("supplementary_data.csv")

# LaTeX table
latex = profile.to_latex()
print(latex)

Batch Analysis

import pandas as pd

models = ["Llama-4-Scout", "DeepSeek-R1", "Qwen3.5-35B", "Gemma-3-12B"]
results = loc.benchmark(models)

# Create DataFrame
rows = []
for p in results.profiles:
    for fs in p.per_function:
        rows.append({
            "model": p.model_id,
            "function": fs.function.value,
            "tc_score": fs.tc_score,
        })

df = pd.DataFrame(rows)
pivot = df.pivot(index="model", columns="function", values="tc_score")
pivot.style.background_gradient(cmap="YlOrRd")