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)
