blood.py 文件源码

def __init__(self, cycle_cap, chain_cap, min, max, w_fun):
        self.cycle_cap = cycle_cap
        self.chain_cap = chain_cap
        self.min = min
        self.max = max
        self.w_fun = w_fun
        self.action_space = spaces.Box(min, max, (len(BLOODS)**2,))

        self.params = {
            "cycle_cap": cycle_cap,
            "chain_cap": chain_cap,
            "min": min,
            "max": max
        }

        self.stats = {
            "cycle_reward": 0,
            "chain_reward": 0
        }

        for blood in BLOODS:
            self.stats["%s_patient_matched" % blood] = 0
            self.stats["%s_donor_matched" % blood] = 0