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

Combine after_load, after_lp and after_mip into Sample dataclass

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-11 17:20:17 -05:00
parent 2d4ded1978
commit fde6dc5a60
4 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/instance/base.py
View File

@@ -8,7 +8,7 @@ from typing import Any, List, Optional, Hashable, TYPE_CHECKING
from overrides import EnforceOverrides
from miplearn.features import TrainingSample, Features
from miplearn.features import TrainingSample, Features, Sample
from miplearn.types import VariableName, Category
logger = logging.getLogger(__name__)
@@ -33,9 +33,7 @@ class Instance(ABC, EnforceOverrides):
def __init__(self) -> None:
self.training_data: List[TrainingSample] = []
self.features: Features = Features()
self.features_after_load: List[Features] = []
self.features_after_lp: List[Features] = []
self.features_after_mip: List[Features] = []
self.samples: List[Sample] = []
@abstractmethod
def to_model(self) -> Any:

4
miplearn/instance/picklegz.py
View File

@@ -123,9 +123,7 @@ class PickleGzInstance(Instance):
self.instance = obj
self.features = self.instance.features
self.training_data = self.instance.training_data
self.features_after_load = self.instance.features_after_load
self.features_after_lp = self.instance.features_after_lp
self.features_after_mip = self.instance.features_after_mip
self.samples = self.instance.samples
@overrides
def free(self) -> None: