def __init__(self, hz=200, sim_speed=5, spawn_rate=60): self.intersection_state = 0 self.state = np.array((self.intersection_state, \ env.northward_cars_waiting, \ env.eastward_cars_waiting, \ env.southward_cars_waiting, \ env.westward_cars_waiting)) self.reward = 0 self.done = False self.intersection = World.Intersection() self.traffic_control = Controller.TrafficController() self.img = pygame.image.load("intersection.png").convert()
def __init__(self, hz=200, sim_speed=1): self.intersection_state = 0 self.hz = hz self.sim_speed = sim_speed self.state = np.array((self.intersection_state, \ env.north_cars_wait_time, \ env.east_cars_wait_time, \ env.south_cars_wait_time, \ env.west_cars_wait_time)) self.reward = 0 self.done = False self.prev_total_wait_time = 0 self.intersection = World.Intersection() self.traffic_control = Controller.TrafficController() self.img = pygame.image.load("intersection.png").convert() self.time = 0