Exemplo n.º 1
0
 def __init__(self, numTran, numStep, volume, time, frunflag):
     self.numSteps = numStep
     self.lastAction = None
     self.lastState = None
     self.env = Env.EnvCls(numTran, frunflag, volume, time)
     self.agent = Agent_base.Agentbase(volume, time)
     self.lastStep = 0
     self.totalRewards = 0
     self.fepisode_over = False
    def __init__(self, numTran, numStep, volume, time, frunflag, parm):
        self.numSteps = numStep
        self.lastAction = None
        self.lastState = None
        self.env = Env.EnvCls(numTran, frunflag, volume, time)
        """Calls separate Agent modules based on different strategies"""
        if parm == "greedy":
            self.agent = Agent.AgentCls(volume, time)
        elif parm == "random":
            self.agent = Agent_rnd.AgentRand(volume, time)
        else:
            self.agent = Agent_baseline.Agentbase(volume, time)

        self.lastStep = 0
        self.totalRewards = 0

        if parm == "greedy" and frunflag == False:
            self.fepisode_over = False
        else:
            self.fepisode_over = True