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()