ddqn_agent.py 文件源码

def agent_message(self, inMessage):
        if inMessage.startswith("freeze learning"):
            self.policyFrozen = True
            return "message understood, policy frozen"

        if inMessage.startswith("unfreeze learning"):
            self.policyFrozen = False
            return "message understood, policy unfrozen"

        if inMessage.startswith("save model"):
            serializers.save_npz('resume.model', self.DDQN.model) # save current model
            np.savez('stored_D012.npz', D0=self.DDQN.D[0], D1=self.DDQN.D[1], D2=self.DDQN.D[2])
            np.savez('stored_D34.npz', D3=self.DDQN.D[3], D4=self.DDQN.D[4])
            return "message understood, model saved"