cog-imperial / OMLT

Represent trained machine learning models as Pyomo optimization formulations
Other
262 stars 57 forks source link

KeyError: 'relu'. #22

Closed wangyexiang closed 2 years ago

wangyexiang commented 2 years ago

Dear sir, I change the ANN model tokeras_linear_131_relu_output_activation(or keras_linear_131_relu), I get the KeyError: 'relu' error. How to fix it? I checked the utils.py file, relu function was not included inpyomo_activations dictionary variable.

EDIT: I use formulation = ReLUBigMFormulation(net), It works.