learning.py 文件源码

def play_net(model, params):
    ws = create_connection("ws://localhost:8000/ws")
    welcomeJson = ws.recv()
    welcome = json.loads(welcomeJson)
    clientId = welcome['snakeID']

    while True:
      state = ws.recv()
      parsed = json.loads(state)

      if parsed['eventType'] != 'World':
        continue

      qval = model.predict(world_json_to_array(parsed['world'], clientId), batch_size=1)
      action = (np.argmax(qval))  # best
      ws.send(json.dumps({"actionType": "Direction", "snakeID": clientId, "direction": intToDirection(action)}))