def __init__(self, seq, strategy = "nn_operator_ext"): self.seq = seq self.action_space = spaces.Discrete(4) # Choose among [0, 1, 2 ,3] self.env = Lattice2D(seq) self.current_sol = [1] self.current_state = self.env._compute_state(self.current_sol) self.strategy = strategy
def __init__(self): self.seq = "HPPHPPH" self.model = FFmodel(8) self.operator = NNOperatorBuilder(self.seq).get("nn_operator") self.env = Lattice2D(self.seq)
def __init__(self, seq): self.seq = seq self.size = (len(seq) - 1) * 4 self.env = Lattice2D(self.seq)