| import torch | |
| def epistemic(outdict): | |
| ''' | |
| Epistemic uncertainty given the output of evidential model | |
| ''' | |
| beta = outdict['beta'] | |
| alpha = outdict['alpha'] | |
| v = outdict['v'] | |
| return (beta / (v * (alpha - 1))).item() | |
| def aleatoric(outdict): | |
| ''' | |
| Aleatoric uncertainty given the output of evidential model | |
| ''' | |
| beta = outdict['beta'] | |
| alpha = outdict['alpha'] | |
| return (beta / (alpha - 1)).item() | |