Skip to content

Commit

Permalink
make action space varies
Browse files Browse the repository at this point in the history
  • Loading branch information
EdenWuyifan committed May 1, 2024
1 parent 6bab136 commit 852e3b3
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -60,3 +60,6 @@ lightning_logs
.cache
**/smac3_output
/venv

.vscode
**/.vscode
2 changes: 1 addition & 1 deletion alpha_automl/pipeline_search/agent_environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,10 +34,10 @@ def __init__(self, config: EnvContext):
), # Ray env board contains pipeline and metadata
}
)
self.max_actions = 24 # max number of actions (depends on the largest step in the grammar, i.e. CLASSIFIER)
self.action_spaces = (
self.generate_action_spaces()
) # number of actions for each step
self.max_actions = max(list(self.action_spaces.values())) # max number of actions (depends on the largest step in the grammar, i.e. CLASSIFIER)
self.action_offsets = (
self.generate_action_offsets()
) # offset for each step, for translating action to PipelineGame action
Expand Down

0 comments on commit 852e3b3

Please sign in to comment.