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

AlvLouWeh2017: Add some assertions; replace non-finite by zero

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-12 07:52:48 -05:00
parent ccb1a1ed25
commit 78d2ad4857
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
miplearn/features/extractor.py
View File

@@ -363,6 +363,9 @@ class FeaturesExtractor:
push(np.abs(c) / c_neg_sum) push(np.abs(c) / c_neg_sum)
if A is not None: if A is not None:
assert A.shape[1] == nvars
assert A.shape[0] == len(b)
M1 = A.T.multiply(1.0 / np.abs(b)).T.tocsr() M1 = A.T.multiply(1.0 / np.abs(b)).T.tocsr()
M1_pos = M1[b > 0, :] M1_pos = M1[b > 0, :]
if M1_pos.shape[0] > 0: if M1_pos.shape[0] > 0:
@@ -426,4 +429,4 @@ def _fix_infinity(m: Optional[np.ndarray]) -> None:
max_values = np.max(masked, axis=0) max_values = np.max(masked, axis=0)
min_values = np.min(masked, axis=0) min_values = np.min(masked, axis=0)
m[:] = np.maximum(np.minimum(m, max_values), min_values) m[:] = np.maximum(np.minimum(m, max_values), min_values)
m[np.isnan(m)] = 0.0 m[~np.isfinite(m)] = 0.0