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

Split Sample.{get,put} into {get,put}_{scalar,vector}

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-07-14 10:50:54 -05:00
parent 0a399deeee
commit 8fc7c6ab71
4 changed files with 165 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/solvers/learning.py
View File

@@ -210,7 +210,7 @@ class LearningSolver:
logger.info("Extracting features (after-lp)...")
initial_time = time.time()
for (k, v) in lp_stats.__dict__.items():
sample.put(k, v)
sample.put_scalar(k, v)
self.extractor.extract_after_lp_features(self.internal_solver, sample)
logger.info(
"Features (after-lp) extracted in %.2f seconds"
@@ -278,7 +278,7 @@ class LearningSolver:
logger.info("Extracting features (after-mip)...")
initial_time = time.time()
for (k, v) in mip_stats.__dict__.items():
sample.put(k, v)
sample.put_scalar(k, v)
self.extractor.extract_after_mip_features(self.internal_solver, sample)
logger.info(
"Features (after-mip) extracted in %.2f seconds"