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

Primal: Add end-to-end tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-31 12:38:23 -05:00
parent db2f426140
commit b83911a91d
3 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/components/primal.py
View File

@@ -86,8 +86,8 @@ class PrimalSolutionComponent(Component):
else:
self._n_one += 1
logger.info(
f"Predicted: {self._n_free} free, {self._n_zero} fix-zero, "
f"{self._n_one} fix-one"
f"Predicted: free: {self._n_free}, zero: {self._n_zero}, "
f"one: {self._n_one}"
)
# Provide solution to the solver
@@ -146,8 +146,8 @@ class PrimalSolutionComponent(Component):
thr = self.thresholds[category].predict(xc)
y_pred[category] = np.vstack(
[
proba[:, 0] > thr[0],
proba[:, 1] > thr[1],
proba[:, 0] >= thr[0],
proba[:, 1] >= thr[1],
]
).T