Build more neural network tests with different activations and edge cases

cog-imperial / OMLT

Represent trained machine learning models as Pyomo optimization formulations

Other

257 stars 56 forks source link

Closed jezsadler closed 2 weeks ago

jezsadler commented 2 weeks ago

Per test_network_definition.py, line 13:

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

jezsadler commented 2 weeks ago

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