Skill

Parameters

class agilerl.wrappers.learning.Skill(env: Env)

The Skill class, used in curriculum learning to teach agents skills. This class works as a wrapper around an environment that alters the reward to encourage learning of a particular skill.

Parameters:

env (Gymnasium-style environment) – Environment to learn in

step(action: Any) tuple[Any, float, bool, bool, dict[str, Any]]

Uses the step() of the env that can be overwritten to change the returned data.

BanditEnv

Parameters

class agilerl.wrappers.learning.BanditEnv(features: DataFrame, targets: DataFrame)

The Bandit learning environment class. Turns a labelled dataset into a reinforcement learning, Gym-style environment.

Parameters:
  • features (Pandas DataFrame) – Dataset features

  • targets – Dataset targets corresponding to features