cog-imperial / OMLT

Represent trained machine learning models as Pyomo optimization formulations
Other
281 stars 59 forks source link

Build more neural network tests with different activations and edge cases #158

Closed jezsadler closed 5 months ago

jezsadler commented 5 months ago

Per test_network_definition.py, line 13:

# TODO @cog-imperial: Build more tests with different activations and edge cases

jezsadler commented 5 months ago

Also, per test_relu.py, line 14 # TODO @cog-imperial: Add tests for single dimensional outputs as well