UserCutsComponent: ignore training instances without cut data

pull/3/head
Alinson S. Xavier 5 years ago
parent 75570ceaeb
commit b51c367257

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

Loading…
Cancel
Save