Source code for stable_baselines3.her.goal_selection_strategy

from enum import Enum

[docs]class GoalSelectionStrategy(Enum): """ The strategies for selecting new goals when creating artificial transitions. """ # Select a goal that was achieved # after the current step, in the same episode FUTURE = 0 # Select the goal that was achieved # at the end of the episode FINAL = 1 # Select a goal that was achieved in the episode EPISODE = 2
# For convenience # that way, we can use string to select a strategy KEY_TO_GOAL_STRATEGY = { "future": GoalSelectionStrategy.FUTURE, "final": GoalSelectionStrategy.FINAL, "episode": GoalSelectionStrategy.EPISODE, }