scml.oneshot.rl.common

Module Contents

Functions

model_wrapper(→ RLModel)

Wraps a stable_baselines3 model as an RL model

Attributes

RLState	We assume that RL states are numpy arrays
RLAction	We assume that RL actions are numpy arrays
RLModel	A policy is a callable that receives a state and returns an action

scml.oneshot.rl.common.RLState

We assume that RL states are numpy arrays

scml.oneshot.rl.common.RLAction

We assume that RL actions are numpy arrays

scml.oneshot.rl.common.RLModel

A policy is a callable that receives a state and returns an action

scml.oneshot.rl.common.model_wrapper(model, deterministic: bool = False) → RLModel

Wraps a stable_baselines3 model as an RL model