def test_objective_single_game(self):
o = Optimizer()
o.load_games([(1, 2, 1, 1)])
v1 = o.create_vars({1: {1: 2200}, 2: {1: 1800}}, [0, -1.01])
v2 = o.create_vars({1: {1: 2200}, 2: {1: 2200}}, [0, -1.01])
v3 = o.create_vars({1: {1: 1800}, 2: {1: 2200}}, [0, -1.01])
(total1, likelihood1, regularization1, smoothness1,
func_hard_reg, _) = o.objective(v1, verbose=True)
self.assertLess(likelihood1, 0)
self.assertTrue(1E-6 < regularization1 < 1)
self.assertEqual(smoothness1, 0)
self.assertTrue(func_hard_reg < 1)
(total2, likelihood2, regularization2, _, _, _) = o.objective(
v2, verbose=True)
self.assertLess(likelihood2, likelihood1)
(total3, likelihood3, regularization3, _, _, _) = o.objective(
v3, verbose=True)
self.assertAlmostEqual(regularization1, regularization3)
self.assertLess(total1 / total2, 0.9)
self.assertLess(total2 / total3, 0.9)
def test_objective_draw(self):
o = Optimizer(rating_reg=0)
o.load_games([(1, 2, 1, 2), (2, 1, 1, 2)])
v1 = o.create_vars({1: {1: 2200}, 2: {1: 1800}}, [0, -1.01])
v2 = o.create_vars({1: {1: 2000}, 2: {1: 2000}}, [0, -1.01])
v3 = o.create_vars({1: {1: 1800}, 2: {1: 2200}}, [0, -1.01])
self.assertLess(o.objective(v2), o.objective(v1))
self.assertLess(o.objective(v2), o.objective(v3))
def test_objective_symmetric_wins(self):
o = Optimizer(rating_reg=0)
o.load_games([(1, 2, 1, 1), (2, 1, 1, 1)])
v1 = o.create_vars({1: {1: 2200}, 2: {1: 1800}}, [0, -1.01])
v2 = o.create_vars({1: {1: 2000}, 2: {1: 2000}}, [0, -1.01])
v3 = o.create_vars({1: {1: 1800}, 2: {1: 2200}}, [0, -1.01])
self.assertLess(o.objective(v2) / o.objective(v1), 0.9)
self.assertLess(o.objective(v2) / o.objective(v3), 0.9)
def test_objective_time_reg(self):
o = Optimizer(rand_seed=239)
o.load_games([(1, 2, 1, 1), (1, 2, 2, 0)])
v = o.create_vars({1: {1: 2200, 2: 1800}, 2: {1: 1800, 2: 2200}},
(0, -1.01))
(total, likelihood, regularization, _, _, _) = o.objective(
v, verbose=True)
self.assertLess(likelihood, 0)
self.assertTrue(1E-6 < regularization < 1)
def test_gradient(self):
o = Optimizer(func_hard_reg=0, func_soft_reg=0, time_delta=0,
rating_reg=0)
o.load_games([(1, 2, 1, 1)])
v = o.create_vars({1: {1: 2200}, 2: {1: 1800}}, (0, -1.01))
def o0(x):
save_x = v[0]
v[0] = x
res = o.objective(v)
v[0] = save_x
return res
def o1(x):
save_x = v[1]
v[1] = x
res = o.objective(v)
v[1] = save_x
return res
self.assertAlmostEqual(derivative(o0, v[0]), o.gradient(v)[0])
self.assertAlmostEqual(derivative(o1, v[1]), o.gradient(v)[1])