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

StaticLazy: Refactor

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-04 08:39:56 -05:00
parent 168f56c296
commit 6e614264b5
8 changed files with 340 additions and 429 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
miplearn/components/objective.py
View File

@@ -58,8 +58,8 @@ class ObjectiveValueComponent(Component):
def fit_xy(
self,
x: Dict[str, np.ndarray],
y: Dict[str, np.ndarray],
x: Dict[Hashable, np.ndarray],
y: Dict[Hashable, np.ndarray],
) -> None:
for c in ["Upper bound", "Lower bound"]:
if c in y:
@@ -84,9 +84,9 @@ class ObjectiveValueComponent(Component):
def sample_xy(
features: Features,
sample: TrainingSample,
) -> Tuple[Dict[str, List[List[float]]], Dict[str, List[List[float]]]]:
x: Dict[str, List[List[float]]] = {}
y: Dict[str, List[List[float]]] = {}
) -> Tuple[Dict[Hashable, List[List[float]]], Dict[Hashable, List[List[float]]]]:
x: Dict[Hashable, List[List[float]]] = {}
y: Dict[Hashable, List[List[float]]] = {}
f = list(features["Instance"]["User features"])
if "LP value" in sample and sample["LP value"] is not None:
f += [sample["LP value"]]