Inhibited-softmax icon indicating copy to clipboard operation
Inhibited-softmax copied to clipboard

Repository with code for paper "Inhibited Softmax for Uncertainty Estimation in Neural Networks"

Results 1 Inhibited-softmax issues
Sort by recently updated
recently updated
newest added

Hi, Excuse my ignorance, I want to ask that in **models_code/experiments.py**, function **test_eval()** has a line `y_.cpu().data.numpy()[:, :num_classes].argmax(axis=1)`. Now my question is that, why are you using `y_.cpu().data.numpy()` why not...