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

expert primal: Set value for int variables

Browse Source
This commit is contained in:
Alinson S. Xavier
2024-05-31 11:48:41 -05:00
parent 26cfab0ebd
commit 7f273ebb70
3 changed files with 31 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/components/primal/expert.py
View File

@@ -5,7 +5,7 @@
import logging
from typing import Any, Dict, List
from . import _extract_bin_var_names_values
from . import _extract_int_var_names_values
from .actions import PrimalComponentAction
from ...solvers.abstract import AbstractModel
from ...h5 import H5File
@@ -28,5 +28,5 @@ class ExpertPrimalComponent:
self, test_h5: str, model: AbstractModel, stats: Dict[str, Any]
) -> None:
with H5File(test_h5, "r") as h5:
names, values, _ = _extract_bin_var_names_values(h5)
names, values, _ = _extract_int_var_names_values(h5)
self.action.perform(model, names, values.reshape(1, -1), stats)