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

Extract all features ahead of time

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-31 07:42:01 -05:00
parent b3c24814b0
commit 12fca1f22b
8 changed files with 106 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
miplearn/instance.py
View File

@@ -12,6 +12,7 @@ import numpy as np
from miplearn.types import TrainingSample, VarIndex, ModelFeatures
# noinspection PyMethodMayBeStatic
class Instance(ABC):
"""
Abstract class holding all the data necessary to generate a concrete model of the
@@ -26,7 +27,7 @@ class Instance(ABC):
def __init__(self) -> None:
self.training_data: List[TrainingSample] = []
self.model_features: ModelFeatures = {}
self.features: ModelFeatures = {}
@abstractmethod
def to_model(self) -> Any:
@@ -94,10 +95,10 @@ class Instance(ABC):
"""
return "default"
def get_constraint_features(self, cid):
return np.zeros(1)
def get_constraint_features(self, cid: str) -> Optional[List[float]]:
return [0.0]
def get_constraint_category(self, cid):
def get_constraint_category(self, cid: str) -> Optional[str]:
return cid
def has_static_lazy_constraints(self):