def __init__(self, model_name): super(SetOptimizer, self).__init__() self.model = ItemSetModel() self.model.load(model_name) self.item_ids = [x.item_id for x in Item.objects().all()] self.item_ids.sort()
def _optimize_build(self): opt = SetOptimizer('RandomForest_updated') res = opt.optimize(72, [4, 1, 1, 1, 1, 1]) for item in res.x: print(Item.objects(item_id=item).first().name)
def _optimize_build(self): opt = SetOptimizer('RandomForest_updated') res = opt.optimize(72, [4,1,1,1,1,1]) for item in res.x: print(Item.objects(item_id=item).first().name)