Skip to content

lit_sb3.core.plots.score_normalization

source module lit_sb3.core.plots.score_normalization

Score normalization utilities for Pybullet environment benchmarks.

Adapted from https://github.com/DLR-RM/rl-baselines3-zoo/blob/master/rl_zoo3/plots/score_normalization.py

Functions

  • normalize_score — Normalize scores to [0, 1] range using min/max for the environment.

source normalize_score(scores: np.ndarray, env_id: str) → np.ndarray

Normalize scores to [0, 1] range using min/max for the environment.

:param scores: Array of scores. :param env_id: Environment ID. :return: Normalised scores. :raises KeyError: If env_id is not in the normalisation tables.