def testLinearlyDecayingEpsilon(self): """Test the functionality of the linearly_decaying_epsilon function.""" decay_period = 100 warmup_steps = 6 epsilon = 0.1 steps_schedule = [ (0, 1.0), # step < warmup_steps (16, 0.91), # bonus = 0.9 * 90 / 100 = 0.81 (decay_period + warmup_steps + 1, epsilon)] # step > decay+warmup for step, expected_epsilon in steps_schedule: self.assertNear(dqn_agent.linearly_decaying_epsilon(decay_period, step, warmup_steps, epsilon), expected_epsilon, 0.01)
def testLinearlyDecayingEpsilon(self): """Test the functionality of the linearly_decaying_epsilon function.""" decay_period = 100 warmup_steps = 6 epsilon = 0.1 steps_schedule = [ (0, 1.0), # step < warmup_steps (16, 0.91), # bonus = 0.9 * 90 / 100 = 0.81 (decay_period + warmup_steps + 1, epsilon)] # step > decay+warmup for step, expected_epsilon in steps_schedule: self.assertNear(dqn_agent.linearly_decaying_epsilon(decay_period, step, warmup_steps, epsilon), expected_epsilon, 0.01)