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

Correctly store features and training data for file-based instances

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-04 22:00:21 -05:00
parent 025e08f85e
commit f2520f33fb
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
miplearn/solvers/learning.py
View File

@@ -135,8 +135,6 @@ class LearningSolver:
# Initialize training sample
training_sample: TrainingSample = {}
if not hasattr(instance, "training_data"):
instance.training_data = []
instance.training_data += [training_sample]
# Initialize stats
@@ -151,7 +149,8 @@ class LearningSolver:
# Extract features
extractor = FeaturesExtractor(self.internal_solver)
instance.features = extractor.extract(instance)
instance.features.clear() # type: ignore
instance.features.update(extractor.extract(instance))
callback_args = (
self,