search

lanl-ansi / MathOptAI.jl

Embed trained machine learning predictors in JuMP
https://lanl-ansi.github.io/MathOptAI.jl/
Other
29 stars 1 forks source link

Add reduced_space keyword argument #71

Closed odow closed 2 months ago

odow commented 2 months ago

One option for #70

odow commented 2 months ago

The other option would be to add a reduced_space::Bool flag to each predictor.

I guess the question is: how fine-grained to people expect these to be? Do I want all layers in a NN to be reduced-space? Only some? If so, which ones? All Affine but not ReLU?

odow commented 2 months ago

A third option would be to add separate predictors, so something like ReducedSpace{ReLU}.

odow commented 2 months ago

Yeah, I think I like https://github.com/lanl-ansi/MathOptAI.jl/pull/72 more.

odow commented 2 months ago

Closing in favor of #72