# create agents agents = [] for i in world.intersections: # define tempo de amarelo yellow_phase_time = i.yellow_phase_time action_space = gym.spaces.Discrete(len(i.phases)) agents.append( TOSFB( action_space, StateOfThreeGenerator(world, i, ["state_of_three"], in_only=True, average=None), LaneVehicleGenerator(world, i, ["lane_waiting_count"], in_only=True, average="all", negative=True), yellow_phase_time, i, world)) if args.load_model: agents[-1].load_model(args.save_dir) wandb.init(project='Explainability', name='tosfb', save_code=True, config={ 'lr': agents[-1].alpha, 'fourier_order': agents[-1].fourier_order, 'gamma': agents[-1].gamma, 'min_epsilon': agents[-1].min_epsilon, 'lambda': agents[-1].lamb, 'max_nonzero_fourier': agents[-1].max_non_zero_fourier,