From b51c367257ad27656fd96235c39fe145c7645262 Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Mon, 5 Oct 2020 11:26:36 -0500 Subject: [PATCH] UserCutsComponent: ignore training instances without cut data --- miplearn/components/cuts.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/miplearn/components/cuts.py b/miplearn/components/cuts.py index 92f4a55..7743fd1 100644 --- a/miplearn/components/cuts.py +++ b/miplearn/components/cuts.py @@ -48,6 +48,8 @@ class UserCutsComponent(Component): self.classifiers = {} violation_to_instance_idx = {} for (idx, instance) in enumerate(training_instances): + if not hasattr(instance, "found_violated_user_cuts"): + continue for v in instance.found_violated_user_cuts: if v not in self.classifiers: self.classifiers[v] = deepcopy(self.classifier_prototype)