oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#1522: refactor red_agent_identifier -> random_red_agent so that it is a boolean + documentation

Browse Source
This commit is contained in:
Czar Echavez
2023-07-03 12:18:58 +01:00
parent 6b4530bded
commit befd183b2c
7 changed files with 10 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/config/_package_data/training/training_config_main.yaml
View File

@@ -9,7 +9,7 @@ agent_identifier: STABLE_BASELINES3_A2C
# RED AGENT IDENTIFIER
# RANDOM or NONE
red_agent_identifier: "NONE"
random_red_agent: False
# Sets How the Action Space is defined:
# "NODE"

2
src/primaite/config/_package_data/training/training_config_random_red_agent.yaml
View File

@@ -9,7 +9,7 @@ agent_identifier: STABLE_BASELINES3_A2C
# RED AGENT IDENTIFIER
# RANDOM or NONE
red_agent_identifier: "RANDOM"
random_red_agent: True
# Sets How the Action Space is defined:
# "NODE"

2
src/primaite/config/training_config.py
View File

@@ -21,7 +21,7 @@ class TrainingConfig:
agent_identifier: str = "STABLE_BASELINES3_A2C"
"The Red Agent algo/class to be used."
red_agent_identifier: str = "RANDOM"
random_red_agent: bool = False
"Creates Random Red Agent Attacks"
action_type: ActionType = ActionType.ANY