about the variance of classification task

kyle-dorman / bayesian-neural-network-blogpost

Building a Bayesian deep learning classifier

481 stars 105 forks source link

Open ShellingFord221 opened 4 years ago

ShellingFord221 commented 4 years ago

Hi, when modeling aleatoric uncertainty, why is pred_var, predicted logit values and variance's shape: (N, C + 1)? Shouldn't it be (N, C * 2)?