ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-08 18:38:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix all tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-25 15:19:58 -06:00
parent 3ab3bb3c1f
commit b0b013dd0a
5 changed files with 48 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/components/steps/convert_tight.py
View File

@@ -116,9 +116,11 @@ class ConvertTightIneqsIntoEqsStep(Component):
if category not in y:
y[category] = []
if 0 <= slack <= self.slack_tolerance:
y[category] += [[1]]
y[category] += [[False, True]]
else:
y[category] += [[0]]
y[category] += [[True, False]]
for category in y.keys():
y[category] = np.array(y[category], dtype=np.bool8)
return y
def predict(self, x):