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

Add mip_stats to after-mip features

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-11 09:14:05 -05:00
parent bd78518c1f
commit 6b15337e4c
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
miplearn/features.py
View File

@@ -13,7 +13,7 @@ import numpy as np
from miplearn.types import Solution, Category
if TYPE_CHECKING:
from miplearn.solvers.internal import InternalSolver, LPSolveStats
from miplearn.solvers.internal import InternalSolver, LPSolveStats, MIPSolveStats
from miplearn.instance.base import Instance
@@ -103,6 +103,7 @@ class Features:
variables: Optional[Dict[str, Variable]] = None
constraints: Optional[Dict[str, Constraint]] = None
lp_solve: Optional["LPSolveStats"] = None
mip_solve: Optional["MIPSolveStats"] = None
class FeaturesExtractor: