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: Remove slow loop in M3

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-13 06:07:46 -05:00
parent f2b710e9f9
commit 9e023a375a
1 changed files with 1 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
miplearn/features/extractor.py
View File

@@ -356,17 +356,8 @@ class FeaturesExtractor:
push(np.abs(v)) push(np.abs(v))
def maxmin(M: np.ndarray) -> Tuple[np.ndarray, np.ndarray]: def maxmin(M: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
# Compute max using regular numpy operations
M_max = np.ravel(M.max(axis=0).todense()) M_max = np.ravel(M.max(axis=0).todense())
M_min = np.ravel(M.min(axis=0).todense())
# Compute min by iterating through the sparse matrix data, so that
# we skip non-zero entries
M_min = np.array(
[
0.0 if len(M[:, j].data) == 0 else M[:, j].data.min()
for j in range(M.shape[1])
]
)
return M_max, M_min return M_max, M_min
with np.errstate(divide="ignore", invalid="ignore"): with np.errstate(divide="ignore", invalid="ignore"):