Skip to content

Visualization

The repository includes visualization helpers for environments, graphs, training curves, shield telemetry, and conceptual diagrams.

Environment Rendering

Environment renderers live beside the environments under src/environments/.

Gridworld renderers show spatial layouts, agents, obstacles, and task-specific state. Matrix renderers give compact visual summaries for repeated matrix-style games.

Reward and Safety Plots

Experiment notebooks export cumulative and episodic plots for reward and safety.

When a workflow includes world-model pretraining, some plots include the pretraining prefix and some start at policy training. This keeps both the full timeline and the post-training comparison available.

Shield Telemetry

Shielded experiments can record:

  • override counts,
  • unsafe proposal counts,
  • missing coverage counts,
  • proposed and final risk statistics.

These summaries help distinguish a policy that is safe on its own from a policy that is safe because the shield frequently intervenes.

Diagram Notebooks

Diagram notebooks live under:

notebooks/diagrams/

They cover world-model graphs, opponent-model reasoning levels, and shielding diagrams. They are useful for explaining the pipeline without running full experiments.