From ab8fdad47fef584313d10ab7e4c7498e4b861ff0 Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Fri, 1 May 2020 09:54:42 -0500 Subject: [PATCH] Add x and y methods to PrimalSolutionComponent --- src/python/miplearn/components/primal.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/python/miplearn/components/primal.py b/src/python/miplearn/components/primal.py index fd15fb2..d430768 100644 --- a/src/python/miplearn/components/primal.py +++ b/src/python/miplearn/components/primal.py @@ -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)