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

Add x and y methods to PrimalSolutionComponent

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-05-01 09:54:42 -05:00
parent e352e478d0
commit ab8fdad47f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/python/miplearn/components/primal.py
View File

@@ -38,6 +38,12 @@ class PrimalSolutionComponent(Component):
def after_solve(self, solver, instance, model, results):
pass
def x(self, training_instances):
return VariableFeaturesExtractor().extract(training_instances)
def y(self, training_instances):
return SolutionExtractor().extract(training_instances)
def fit(self, training_instances, n_jobs=1):
logger.debug("Extracting features...")
features = VariableFeaturesExtractor().extract(training_instances)