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

Implement PrimalSolutionComponent; remove deprecated predictors

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-02-24 10:01:21 -06:00
parent ccd694af9b
commit a2fbb9f8d8
12 changed files with 140 additions and 669 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
miplearn/extractors.py
View File

@@ -42,8 +42,8 @@ class Extractor(ABC):
results[category] = np.hstack(results[category])
return results
class UserFeaturesExtractor(Extractor):
class VariableFeaturesExtractor(Extractor):
def extract(self,
instances,
models=None,
@@ -62,6 +62,7 @@ class UserFeaturesExtractor(Extractor):
result[category] += [np.hstack([
instance_features,
instance.get_variable_features(var, index),
instance.lp_solution[str(var)][index],
])]
for category in result.keys():
result[category] = np.vstack(result[category])