def __init__(self): self.DataShape = (None, config.MAP_SIZE, config.MAP_SIZE, 39) self.controller = decision_maker( DQN(config.MU, config.SIGMA, config.LEARING_RATE, len(sa.build_controller), 5, self.DataShape, 'build_controller')) self.index = handcraft_function.find_controller_index( sa.build_controller)
def __init__(self): # self.DataShape = (None, 200, config.FEATURE_UNITS_LENGTH, 1) self.DataShape = (None, config.MAP_SIZE, config.MAP_SIZE, 39) self.controller = decision_maker( DQN(config.MU, config.SIGMA, config.LEARING_RATE, len(sa.attack_controller), config.ATTACT_CONTROLLER_PARAMETERDIM, self.DataShape, 'attack_controller')) self.index = handcraft_function.find_controller_index( sa.attack_controller)