# 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,