Open manuelgloeckler opened 7 months ago
I find the SNPE_A_MDN
class quite confusing. Is there a loss implemented for it yet? Why does it only inherit from DensityEstimator
but not implement all methods?
This is being worked on in #1042 but it will not make it into the release in August.
Description
The MDN i.e. Gaussian Mixture Density network, is a density estimator which gets a lot of "special" treatment.
SNPE_A_MDN
wrapper.conditional_density.py
implements a specialConditionedMDN
to use the "marginalization" properties.Hence it may be good to collapse all current MDN wrappers/functionalities into a single class
MDNDensityEstimator
, which does follow the newDensityEstimator
.Advantages: