Hi @Ribeiro!
Not sure I can help a lot but for what it’s worth, before being a Flux.jl question, it’s a math research question. A keyword you might use is “feature importance”. So I’m not sure you’ll find a lot of things ready to use, but there are some packages that can do it: