Irwin.py 文件源码

def buildPlayerGameActivationsTable(self, model=None):
    if model is None:
      print("using default model")
      model = self.narrowGameModel.model()
    print("getting players")
    engines = self.env.playerDB.byEngine(True)
    legits = self.env.playerDB.byEngine(False)

    print("got " + str(len(engines + legits)) + " players")

    playerGameActivations = []

    for player in engines + legits:
      print("predicting " + player.id)
      gs = GameAnalysisStore.new()
      gs.addGameAnalyses(self.env.gameAnalysisDB.byUserId(player.id))
      predictions = self.predict(gs.gameAnalysisTensors(), model)
      playerGameActivations.append(PlayerGameActivations(player.id, player.engine, [int(100*np.asscalar(p[0][0][0])) for p in predictions]))

    print("writing to DB")
    self.env.playerGameActivationsDB.lazyWriteMany(playerGameActivations)