def test_learn_dense_cb_explore_adf_2(self):
actual = VowpalLearner._learn_format(True, 0.33, [(1, 0, 0), (0, 1, 0),
(0, 0, 1)],
(1, 2, 3), (0, 0, 1), 0.25)
expected = "shared |s 0:1 1:2 2:3\n|a 0:1\n|a 1:1\n0:-0.25:0.33 |a 2:1"
self.assertEqual(actual, expected)
def test_learn_sparse_cb_explore_2(self):
actual = VowpalLearner._learn_format(False, 0.33,
[(1, 0, 0), (0, 1, 0), (0, 0, 1)],
((1, 5), (3, 4)), (1, 0, 0), .5)
expected = "1:-0.5:0.33 |s 1:3 5:4"
self.assertEqual(actual, expected)
def test_learn_sparse_cb_explore_adf_1(self):
actual = VowpalLearner._learn_format(True, 0.11, ["A", "B", "C"], {
"D": 1,
"E": 2
}, "C", 0.33)
expected = "shared |s D:1 E:2\n|a A:1\n|a B:1\n0:-0.33:0.11 |a C:1"
self.assertEqual(actual, expected)
def test_learn_dense_cb_explore_2(self):
actual = VowpalLearner._learn_format(False,
0.33, [(1, 0, 0), (0, 1, 0),
(0, 0, 1)], (1, 2, 3),
(1, 0, 0), .5)
expected = "1:-0.5:0.33 |s 0:1 1:2 2:3"
self.assertEqual(actual, expected)
def test_learn_dense_cb_explore_1(self):
actual = VowpalLearner._learn_format(False,
0.25, [(1, 0, 0), (0, 1, 0),
(0, 0, 1)], (1, 2, 3),
(0, 1, 0), 1)
expected = "2:-1:0.25 |s 0:1 1:2 2:3"
self.assertEqual(actual, expected)
def test_learn_sparse_cb_explore_1(self):
actual = VowpalLearner._learn_format(False,
0.33, [(1, 0, 0), (0, 1, 0),
(0, 0, 1)], {
"A": 1,
"B": 2
}, (1, 0, 0), .5)
expected = "1:-0.5:0.33 |s A:1 B:2"
self.assertEqual(actual, expected)
def test_learn_sparse_cb_explore_adf_2(self):
actual = VowpalLearner._learn_format(True, 0.11, ["A", "B", "C"],
((3, 5), (2, 3)), "C", 0.33)
expected = "shared |s 3:2 5:3\n|a A:1\n|a B:1\n0:-0.33:0.11 |a C:1"
self.assertEqual(actual, expected)
