csle_agents.agents.base package

Submodules

csle_agents.agents.base.base_agent module

class csle_agents.agents.base.base_agent.BaseAgent(simulation_env_config: SimulationEnvConfig, emulation_env_config: Optional[EmulationEnvConfig], experiment_config: ExperimentConfig, create_log_dir: bool = True)[source]

Bases: ABC

Abstract class representing an RL agent

abstract hparam_names() → List[str][source]

Abstract method to be implemented by subclasses. Gets the list of hyperparameters.

Returns

the list of hyperparameters

abstract train() → ExperimentExecution[source]

Abstract method to be implemented by subclasses. Should contain the training logic

Returns

the training result

Module contents