This commit is contained in:
Kylin_on_Mac
2022-05-26 10:02:31 +08:00
parent 3eb14e009d
commit 458bf2fa46
5 changed files with 21 additions and 86 deletions

View File

@ -1,38 +0,0 @@
import random
import pygame
class MLPlay:
def __init__(self):
print("Initial ml script")
def update(self, scene_info: dict, keyboard:list=[], *args, **kwargs):
"""
Generate the command according to the received scene information
"""
# print("AI received data from game :", json.dumps(scene_info))
# print(scene_info)
actions = []
if pygame.K_w in keyboard or pygame.K_UP in keyboard:
actions.append("UP")
elif pygame.K_s in keyboard or pygame.K_DOWN in keyboard:
actions.append("DOWN")
elif pygame.K_a in keyboard or pygame.K_LEFT in keyboard:
actions.append("LEFT")
elif pygame.K_d in keyboard or pygame.K_RIGHT in keyboard:
actions.append("RIGHT")
else:
actions.append("NONE")
return actions
def reset(self):
"""
Reset the status
"""
print("reset ml script")
pass

View File

@ -1,21 +0,0 @@
import random
class MLPlay:
def __init__(self):
print("Initial ml script")
def update(self, scene_info: dict, *args, **kwargs):
"""
Generate the command according to the received scene information
"""
# print("AI received data from game :", json.dumps(scene_info))
# print(scene_info)
actions = ["UP", "DOWN", "LEFT", "RIGHT"]
return random.sample(actions, 1)
def reset(self):
"""
Reset the status
"""
print("reset ml script")
pass