neurostatslab / vocalocator

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

Add temperature scaling parameter to ProbabilisticOutput #31

Closed amanchoudhri closed 1 year ago

amanchoudhri commented 1 year ago

Currently no way to easily evaluate models with temperature scaled output. I suggest adding such a temperature kwarg to the pmf method of ProbabilisticOutput, and giving the user the ability to specify a constant temperature choice in assess.py from the command line.

Motivation: [Guo et al, 2017]