dodox/mujoco_maze
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Experimental] NoRewardRoom-v0

Browse Source
This commit is contained in:
kngwyu 2021-04-11 18:37:10 +09:00
parent 1295175f9c
commit 72fa14d786
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
mujoco_maze/maze_task.py
View File

@ -329,6 +329,29 @@ class SubGoalTRoom(GoalRewardTRoom):
) )
class NoRewardRoom(MazeTask):
REWARD_THRESHOLD: float = 0.0
MAZE_SIZE_SCALING: Scaling = Scaling(4.0, 4.0, 4.0)
def reward(self, obs: np.ndarray) -> float:
return 0.0
@staticmethod
def create_maze() -> List[List[MazeCell]]:
E, B, R = MazeCell.EMPTY, MazeCell.BLOCK, MazeCell.ROBOT
return [
[B, B, B, B, B, B, B, B, B],
[B, E, E, B, E, E, E, E, B],
[B, E, E, B, E, E, E, E, B],
[B, E, E, E, E, E, B, B, B],
[B, E, E, E, R, E, E, E, B],
[B, B, B, E, E, E, E, E, B],
[B, E, E, E, E, B, E, E, B],
[B, E, E, E, E, B, E, E, B],
[B, B, B, B, B, B, B, B, B],
]
class GoalRewardBlockMaze(GoalRewardUMaze): class GoalRewardBlockMaze(GoalRewardUMaze):
MAZE_SIZE_SCALING: Scaling = Scaling(8.0, 4.0, None) MAZE_SIZE_SCALING: Scaling = Scaling(8.0, 4.0, None)
OBSERVE_BLOCKS: bool = True OBSERVE_BLOCKS: bool = True
@ -460,6 +483,7 @@ class TaskRegistry:
"4Rooms": [DistReward4Rooms, GoalReward4Rooms, SubGoal4Rooms], "4Rooms": [DistReward4Rooms, GoalReward4Rooms, SubGoal4Rooms],
"TRoom": [DistRewardTRoom, GoalRewardTRoom, SubGoalTRoom], "TRoom": [DistRewardTRoom, GoalRewardTRoom, SubGoalTRoom],
"BlockMaze": [DistRewardBlockMaze, GoalRewardBlockMaze], "BlockMaze": [DistRewardBlockMaze, GoalRewardBlockMaze],
"NoRewardRoom": [NoRewardRoom],
"Billiard": [ "Billiard": [
DistRewardBilliard, DistRewardBilliard,
GoalRewardBilliard, GoalRewardBilliard,