csle_agents.agents.base package

Submodules

csle_agents.agents.base.base_agent module

class csle_agents.agents.base.base_agent.BaseAgent(simulation_env_config: csle_common.dao.simulation_config.simulation_env_config.SimulationEnvConfig, emulation_env_config: Optional[csle_common.dao.emulation_config.emulation_env_config.EmulationEnvConfig], experiment_config: csle_common.dao.training.experiment_config.ExperimentConfig)[source]

Bases: abc.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() → csle_common.dao.training.experiment_execution.ExperimentExecution[source]

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

Returns

the training result

Module contents