Model zwracający rozkład zamiast skalaru

Można użyć Monte Carlo Dropout, Bayesian Neural Network.