def setUp(self):
self.le = LogisticEnsemble(models=[IdModel()])
inp = [{
"user": 10,
"item": 100
}, {
"user": 10,
"item": 110
}, {
"user": 12,
"item": 120
}]
self.df = pd.DataFrame(inp)
class LogisticTest(unittest.TestCase):
def setUp(self):
self.le = LogisticEnsemble(models=[IdModel()])
inp = [{
"user": 10,
"item": 100
}, {
"user": 10,
"item": 110
}, {
"user": 12,
"item": 120
}]
self.df = pd.DataFrame(inp)
def test_construct_features(self):
self.le._prepare_feature_extraction(self.df)
self.assertEqual(self.le._user_count, {10: 2, 12: 1})
def test_extract_features(self):
self.le._prepare_feature_extraction(self.df)
features = self.le._extract_features(10, 5)
self.assertEqual(features, ([2, 5], 1))
def test_prepare_data(self):
X, Y = self.le.prepare_data(self.df)
self.assertEqual(len(X), 6)
self.assertEqual(len(Y), 6)
def test_fit(self):
self.le.fit(self.df)
self.assertIsNotNone(self.le.model)
def test_predict(self):
self.le.fit(self.df)
self.assertIsInstance(self.le.get_score(10, 110), float)
class LogisticTest(unittest.TestCase):
def setUp(self):
self.le = LogisticEnsemble(models=[IdModel()])
inp = [{"user": 10, "item": 100}, {"user": 10,"item": 110},
{"user": 12, "item": 120}]
self.df = pd.DataFrame(inp)
def test_construct_features(self):
self.le._prepare_feature_extraction(self.df)
self.assertEqual(self.le._user_count, {10: 2, 12: 1})
def test_extract_features(self):
self.le._prepare_feature_extraction(self.df)
features = self.le._extract_features(10, 5)
self.assertEqual(features, ([2, 5], 1))
def test_prepare_data(self):
X, Y = self.le.prepare_data(self.df)
self.assertEqual(len(X), 6)
self.assertEqual(len(Y), 6)
def test_fit(self):
self.le.fit(self.df)
self.assertIsNotNone(self.le.model)
def test_predict(self):
self.le.fit(self.df)
self.assertIsInstance(self.le.get_score(10, 110), float)
def setUp(self):
self.le = LogisticEnsemble(models=[IdModel()])
inp = [{"user": 10, "item": 100}, {"user": 10,"item": 110},
{"user": 12, "item": 120}]
self.df = pd.DataFrame(inp)
def setUp(self):
self.le = LogisticEnsemble(models=[IdModel()])
inp = [{'user': 10, 'item': 100}, {'user': 10,'item': 110},
{'user': 12,'item': 120}]
self.df = pd.DataFrame(inp)