neurostatslab / vocalocator

Deep neural networks for sound source localization and vocalization attribution.
MIT License
2 stars 0 forks source link

Use weighted average for MDN point estimation #44

Closed amanchoudhri closed 1 year ago

amanchoudhri commented 1 year ago

Currently, the MDNOutput class just takes the unweighted average of all its constituent response distribution's point estimates. This doesn't make any sense, since a MDN also outputs weights that mix the response distributions. Leads to weird behavior like one individual gaussian getting most of the weight, but the point estimate being VERY far from that gaussian.

image