Fix all tests

This commit is contained in:
2021-01-25 15:19:58 -06:00
parent 3ab3bb3c1f
commit b0b013dd0a
5 changed files with 48 additions and 14 deletions

View File

@@ -85,8 +85,10 @@ class DynamicLazyConstraintsComponent(Component):
disable=not sys.stdout.isatty(),
):
logger.debug("Training: %s" % (str(v)))
label = np.zeros(len(training_instances))
label[violation_to_instance_idx[v]] = 1.0
label = [[True, False] for i in training_instances]
for idx in violation_to_instance_idx[v]:
label[idx] = [False, True]
label = np.array(label, dtype=np.bool8)
classifier.fit(features, label)
def predict(self, instance):