-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathdemo.py
30 lines (28 loc) · 967 Bytes
/
demo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from bubble_shooter.process_supervisors.gameplay import GameplaySupervisor
from bubble_shooter.state_preprocessors.all_color import AllColor as AllColorPreprocessor
from bubble_shooter.models.dueling_inception import DuelingInception
TOTAL_PLAYERS = 1
configurations = [
{
'epsilon': 0.99,
'gamma': 0.9,
'learning_rate': 0.00025,
'replay_frequency': 4,
'target_update_frequency': 1000,
'memory_epsilon': 0.01,
'memory_alpha': 0.6,
'memory_size': 10000,
'batch_size': 32,
'episodes': 500,
'steps': 500,
'agent_move_size': 35,
'game_board_width': 560,
'state_preprocessor': AllColorPreprocessor(),
'model_builder': DuelingInception,
},
]
if __name__ == '__main__':
for config in configurations:
supervisor = GameplaySupervisor(config, total_players=TOTAL_PLAYERS)
supervisor.start()
supervisor.wait_to_finish()