csd4ni3l/fleet-commander
1
0
Fork 0
mirror of https://github.com/csd4ni3l/fleet-commander.git synced 2026-01-01 04:23:47 +01:00
Code Issues Packages Projects Releases 1 Wiki Activity

Remove default model i will add back later, add back train.py, fix some rewarding

Browse Source
This commit is contained in:
csd4ni3l
2025-11-16 22:34:29 +01:00
parent 5e87b30f78
commit c7c22695e5
4 changed files with 39 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
utils/rl.py
View File

@@ -172,13 +172,13 @@ class SpaceInvadersEnv(gym.Env):
b = Bullet(self.player.center_x, self.player.center_y, 1)
self.bullets.append(b)
else:
reward -= 0.05
reward -= 0.02
if self.enemy_formation.enemies:
nearest = self._nearest_enemy()
alignment = abs(nearest.center_x - self.player.center_x) / self.width
if alignment < 0.025:
reward += 0.3
reward += 0.1
self.player.center_x = np.clip(self.player.center_x, 0, self.width)
self.player_speed = (self.player.center_x - prev_x) / max(1e-6, PLAYER_SPEED)
@@ -222,7 +222,6 @@ class SpaceInvadersEnv(gym.Env):
if self.player_respawns_remaining > 0:
self.player_respawns_remaining -= 1
self._respawn_player()
reward += 2.0
else:
terminated = True
@@ -265,7 +264,7 @@ class SpaceInvadersEnv(gym.Env):
if self.player.center_x < edge_threshold or self.player.center_x > self.width - edge_threshold:
reward -= 0.03
reward -= 0.005
reward -= 0.0025
obs = self._obs()