Store variables modeling exponential and sum of exponential

Gurobi · Oct 31, 2023 · b6e2333 · b6e2333
1 parent a130501
commit b6e2333
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/gurobi_ml/sklearn/logistic_regression.py b/src/gurobi_ml/sklearn/logistic_regression.py
@@ -255,7 +255,9 @@ def _multi_class_model(self, **kwargs):
             return
         if self.output_type == "probability":
             exp_vars = self.gp_model.addMVar(outputvars.shape)
+            self.exp_vars = exp_vars
             sum_vars = self.gp_model.addMVar((outputvars.shape[0]), lb=self.epsilon)
+            self.sum_vars = sum_vars
             num_gc = self.gp_model.NumGenConstrs
             for index in np.ndindex(outputvars.shape):
                 self.gp_model.addGenConstrExp(