def __init__(self, random_seed=RANDOM_SEED): np.random.seed(random_seed) all_cooked_time, all_cooked_bw, _ = load_trace.load_trace() self.net_env = abrenv.Environment(all_cooked_time=all_cooked_time, all_cooked_bw=all_cooked_bw, random_seed=random_seed) self.last_bit_rate = DEFAULT_QUALITY self.buffer_size = 0. self.state = np.zeros((S_INFO, S_LEN)) self.reset()
def __init__(self, random_seed=RANDOM_SEED): np.random.seed(random_seed) # self.action_space = spaces.Box( # low=0., high=60., shape=(2,), dtype=np.float32) # self.observation_space = spaces.Box( # 0, 10.0, (S_LEN * S_INFO,), dtype=np.float32) all_cooked_time, all_cooked_bw, _ = load_trace.load_trace() self.net_env = abrenv.Environment(all_cooked_time=all_cooked_time, all_cooked_bw=all_cooked_bw, random_seed=RANDOM_SEED) self.last_bit_rate = DEFAULT_QUALITY self.buffer_size = 0. self.state = np.zeros((S_INFO, S_LEN)) self.reset()