search

cog-imperial / OMLT

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

ReducedSpaceNNFormulation: add checking on layer type #155

Open jezsadler opened 3 months ago

jezsadler commented 3 months ago

The ReducedSpaceNNFormulation class has a TODO note to check layer types when building formulation.

I need to know what layer types we support in order to implement this.

jezsadler commented 3 months ago

Just Dense layers with continuous formulations, not Big-M or partition.

jezsadler commented 3 months ago

Added check for Dense layers in my branch, will close on integration.