/rl_visualizer

Primary LanguagePythonOtherNOASSERTION

RL Visualizer

RL Visualizer is a Python package for visualizing action probabilities, entropy, and rewards in reinforcement learning models.

Features

  • Visualize action probabilities as bar charts
  • Track entropy to monitor model uncertainty
  • Plot per-step rewards within a single episode
  • Compatible with Gym and Gymnasium environments

Installation

pip install rl-visualizer