AGENTS.md@src/gnn/analysis/numpyro · git:20260906.e62b9e9 · 2026-09-06 · sha256 964c02fcb1858c66

AGENTS.md@src/gnn/analysis/numpyro git:20260906.e62b9e9A

Immutable. This exact content is served forever at /api/v1/blob/964c02fcb1858c66.

# NumPyro Analysis Sub-module

## Overview

Framework-specific analysis module for NumPyro simulation outputs. Reads `simulation_results.json` produced by the NumPyro runner and generates belief trajectory, action distribution, and Expected Free Energy (EFE) analysis plots.

## Architecture

```
numpyro/
├── __init__.py      # Package exports
└── analyzer.py      # NumPyro result analysis and visualization (202 lines)
```

## Key Functions

- **`generate_analysis_from_logs(results_dir, output_dir=None, verbose=False)`** — Main entry point; reads NumPyro JSON results and produces matplotlib visualizations.
- **Belief trajectory plotting** — Tracks posterior belief evolution over simulation timesteps.
- **Action distribution analysis** — Visualizes policy distributions across actions.

## Dependencies

- `numpy`, `matplotlib` (required)
- NumPyro (runtime, for producing the results analyzed here)

## Parent Module

See [analysis/AGENTS.md](../AGENTS.md) for the overall analysis architecture.

**Version**: 3.2.0