search

sc8668 / RTMScore

MIT License
86 stars 17 forks source link

Additive terms in MDN outputs #7

Closed pmorerio closed 2 years ago

pmorerio commented 2 years ago

https://github.com/sc8668/RTMScore/blob/79475c4c1b68219a0845fdeaf7cfc73600513705/RTMScore/model/model2.py#L531 Hello, thanks for sharing such great code!

What is the meaning of the +1.1 and +1 in the the output for sigma and mu in the mixture density network? Is it some kind of prior knowledge you incorporate in the model? Or simply some numeric regularization?

Thanks in advance for you answer.

P.

sc8668 commented 2 years ago

The MDN module here is just inspired by DeepDock (https://github.com/OptiMaL-PSE-Lab/DeepDock). I guess they are just some numeric regularizations because sigma and mu represent the standard deviations and means of several Gaussians, respectively.

pmorerio commented 2 years ago

Thanks for the quick answer