search

cog-imperial / OMLT

Represent trained machine learning models as Pyomo optimization formulations
Other
273 stars 58 forks source link

Add checks for unsupported activation operation types #25

Open carldlaird opened 2 years ago

carldlaird commented 2 years ago

onnx_parser.py (and possibly keras_reader.py) does not contain checks for unsupported operation types:

=== from PR comment

will this use a linear activation if maybe_node.op_type is not in _ACTIVATION_OP_TYPES? this could lead to an incorrect neural network

Originally posted by @jalving in https://github.com/cog-imperial/OMLT/pull/24#r784097245