Is it working with Keras

pobonomo · pobonomo · commit 172c02a68094 · 2024-12-04T16:07:34.000+01:00
diff --git a/src/gurobi_ml/keras/keras.py b/src/gurobi_ml/keras/keras.py
@@ -110,7 +110,7 @@ def _mip_model(self, **kwargs):
                 output = self._output
             if isinstance(step, keras.layers.InputLayer):
                 pass
-            elif isinstance(step, keras.layers.ReLU):
+            elif isinstance(step, (keras.layers.ReLU, keras.layers.Softmax)):
                 layer = self._add_activation_layer(
                     _input, self.act_dict["relu"](), output, name=f"relu{i}", **kwargs
                 )

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def _mip_model(self, **kwargs):`
`110`	`110`	`output = self._output`
`111`	`111`	`if isinstance(step, keras.layers.InputLayer):`
`112`	`112`	`pass`
`113`		`- elif isinstance(step, keras.layers.ReLU):`
	`113`	`+ elif isinstance(step, (keras.layers.ReLU, keras.layers.Softmax)):`
`114`	`114`	`layer = self._add_activation_layer(`
`115`	`115`	`_input, self.act_dict["relu"](), output, name=f"relu{i}", **kwargs`
`116`	`116`	`)`