cog-imperial / OMLT

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

Enhanced CNN Support #63

Closed pulsipher closed 2 years ago

pulsipher commented 2 years ago

Currently, 2D convolutional layers can be read into OMLT. To better support CNNs, the following layers should also be accommodated:

These additions would enable the use of typical 1D and 2D CNNs (maybe 3D if desired...).