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

GurobiModel: Capture static_var_obj_coeffs_quad

Browse Source
This commit is contained in:
Alinson S. Xavier
2025-06-11 13:19:36 -05:00
parent 1c6912cc51
commit 2ca2794457
3 changed files with 82 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
miplearn/solvers/gurobi.py
View File

@@ -264,6 +264,13 @@ class GurobiModel(AbstractModel):
h5.put_array(
h5_field, np.array(self.inner.getAttr(gp_field, gp_vars), dtype=float)
)
obj = self.inner.getObjective()
if isinstance(obj, gp.QuadExpr):
nvars = len(self.inner.getVars())
obj_q = np.zeros((nvars, nvars))
for i in range(obj.size()):
obj_q[obj.getVar1(i).index, obj.getVar2(i).index] = obj.getCoeff(i)
h5.put_array("static_var_obj_coeffs_quad", obj_q)
def _extract_after_load_constrs(self, h5: H5File) -> None:
gp_constrs = self.inner.getConstrs()