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

Make LearningSolver.add internal

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-19 22:32:05 -06:00
parent 4b8672870a
commit 9ddb952db0
3 changed files with 6 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
miplearn/solvers/learning.py
View File

@@ -104,12 +104,12 @@ class LearningSolver:
if components is not None:
for comp in components:
self.add(comp)
self._add_component(comp)
else:
self.add(ObjectiveValueComponent())
self.add(PrimalSolutionComponent())
self.add(DynamicLazyConstraintsComponent())
self.add(UserCutsComponent())
self._add_component(ObjectiveValueComponent())
self._add_component(PrimalSolutionComponent())
self._add_component(DynamicLazyConstraintsComponent())
self._add_component(UserCutsComponent())
assert self.mode in ["exact", "heuristic"]
for component in self.components.values():
@@ -332,7 +332,7 @@ class LearningSolver:
for component in self.components.values():
component.fit(training_instances)
def add(self, component):
def _add_component(self, component):
name = component.__class__.__name__
self.components[name] = component