aamini/evidential-deep-learning

Possible error?

Emmanuel-Messulam opened this issue · 0 comments

Emmanuel-Messulam commented

evidential-deep-learning/evidential_deep_learning/losses/discrete.py

Line 25 in d1d8e39

B = tf.reduce_sum(alpha*(S-alpha)/(S*S*(S+1)), axis=1, keepdims=True)

Isn't it B = tf.reduce_sum(alpha*(S-alpha)/(S*S*(alpha+1)), axis=1, keepdims=True) from https://chcorbi.github.io/posts/2020/11/dirichlet-networks/#prior-networks ecuation (31)?