agents
Attributes
Classes
Base class for all agents in the Rambling Realms trading system. |
Module Contents
- agents.AGENT_REGISTRY: dict[str, Any]
- class agents.Agent(config, env, data_config=None, load=False)
Base class for all agents in the Rambling Realms trading system. This class provides a common interface and basic functionality for all agents.
- Parameters:
config (trading.cli.alg.config.AgentConfig)
env (gymnasium.Env)
data_config (trading.cli.alg.config.DataConfig | None)
load (bool)
- meta_data: dict
- env: gymnasium.Env
- classmethod make_agent(config, env)
Creates an agent based on the provided configuration and environment.
- Parameters:
config (trading.cli.alg.config.AgentConfig)
env (gymnasium.Env)
- Return type:
stable_baselines3.A2C | stable_baselines3.DDPG | stable_baselines3.DQN | stable_baselines3.PPO | stable_baselines3.SAC
- classmethod load_agent(config, env)
Loads an agent and its meta_data from a saved zip file without extracting to disk. :returns: model, meta_data
- Parameters:
config (trading.cli.alg.config.AgentConfig | pathlib.Path)
env (gymnasium.Env | None)
- learn(timesteps=None, callbacks=None)
Train the agent with enhanced telemetry tracking.
- Parameters:
timesteps (Optional[int]) – Number of timesteps to train for (uses config if None)
callbacks (Optional[list]) – Additional callbacks to include
- Return type:
stable_baselines3.A2C | stable_baselines3.DDPG | stable_baselines3.DQN | stable_baselines3.PPO | stable_baselines3.SAC | Any
- predict(obs)
- save(path=None)
- Parameters:
path (Optional[str])