UncertainPrediction
- class UncertainPrediction(score: torch.FloatTensor, uncertainty: torch.FloatTensor)[source]
Bases:
tuple
A pair of predicted scores and corresponding uncertainty.
Since the uncertainty scores come from Monte-Carlo dropout, they are guaranteed to be non-negative with larger scores indicating higher uncertainty.
Create new instance of UncertainPrediction(score, uncertainty)