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

Increased Aux-Rewards (last decrease was to much)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-07-21 09:41:54 +02:00
parent a9b53b6eca
commit cb403737f8
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
columbus/env.py
View File

@ -381,7 +381,7 @@ class ColumbusCandyland(ColumbusEnv):
class ColumbusCandyland_Aux10(ColumbusCandyland): class ColumbusCandyland_Aux10(ColumbusCandyland):
def __init__(self, fps=30): def __init__(self, fps=30):
super(ColumbusCandyland_Aux10, self).__init__(fps=fps) super(ColumbusCandyland_Aux10, self).__init__(fps=fps)
self.aux_reward_max = 1 self.aux_reward_max = 10
class ColumbusEasyObstacles(ColumbusEnv): class ColumbusEasyObstacles(ColumbusEnv):
@ -389,7 +389,7 @@ class ColumbusEasyObstacles(ColumbusEnv):
super(ColumbusEasyObstacles, self).__init__( super(ColumbusEasyObstacles, self).__init__(
observable=observable, fps=fps, env_seed=env_seed) observable=observable, fps=fps, env_seed=env_seed)
self.draw_entities = not hide_map self.draw_entities = not hide_map
self.aux_reward_max = 1 self.aux_reward_max = 10
def setup(self): def setup(self):
self.agent.pos = self.start_pos self.agent.pos = self.start_pos
@ -412,7 +412,7 @@ class ColumbusEasierObstacles(ColumbusEnv):
super(ColumbusEasierObstacles, self).__init__( super(ColumbusEasierObstacles, self).__init__(
observable=observable, fps=fps, env_seed=env_seed) observable=observable, fps=fps, env_seed=env_seed)
self.draw_entities = not hide_map self.draw_entities = not hide_map
self.aux_reward_max = 1 self.aux_reward_max = 10
def setup(self): def setup(self):
self.agent.pos = self.start_pos self.agent.pos = self.start_pos
@ -435,7 +435,7 @@ class ColumbusJustState(ColumbusEnv):
def __init__(self, observable=observables.StateObservable(), fps=30, num_enemies=0, num_rewards=1, env_seed=None): def __init__(self, observable=observables.StateObservable(), fps=30, num_enemies=0, num_rewards=1, env_seed=None):
super(ColumbusJustState, self).__init__( super(ColumbusJustState, self).__init__(
observable=observable, fps=fps) observable=observable, fps=fps)
self.aux_reward_max = 1 self.aux_reward_max = 10
self.num_enemies = num_enemies self.num_enemies = num_enemies
self.num_rewards = num_rewards self.num_rewards = num_rewards
@ -455,7 +455,7 @@ class ColumbusStateWithBarriers(ColumbusEnv):
def __init__(self, observable=observables.StateObservable(coordsAgent=True, speedAgent=False, coordsRelativeToAgent=False, coordsRewards=True, rewardsWhitelist=None, coordsEnemys=True, enemysWhitelist=None, enemysNoBarriers=True, rewardsTimeouts=False, include_rand=True), fps=30, env_seed=3.141, num_enemys=0, num_barriers=3): def __init__(self, observable=observables.StateObservable(coordsAgent=True, speedAgent=False, coordsRelativeToAgent=False, coordsRewards=True, rewardsWhitelist=None, coordsEnemys=True, enemysWhitelist=None, enemysNoBarriers=True, rewardsTimeouts=False, include_rand=True), fps=30, env_seed=3.141, num_enemys=0, num_barriers=3):
super(ColumbusStateWithBarriers, self).__init__( super(ColumbusStateWithBarriers, self).__init__(
observable=observable, fps=fps, env_seed=env_seed) observable=observable, fps=fps, env_seed=env_seed)
self.aux_reward_max = 1 self.aux_reward_max = 10
self.start_pos = (0.5, 0.5) self.start_pos = (0.5, 0.5)
self.num_barriers = num_barriers self.num_barriers = num_barriers
self.num_enemys = num_enemys self.num_enemys = num_enemys