Ensemble of neural networks with a Gaussian loss function that predict both the mean value and the uncertainty around that mean. See ./tutorials/tutorial_ScatterNN.ipynb for an example.
If you use or find useful any of the code implemented here please cite [1].
The code can be locally pip-installed via
pip install .when in scatternn repository.
[1] Richard Stiskalek; Deaglan J. Bartlett; Harry Desmond; Dhayaa Anbajagane. "The scatter in the galaxy-halo connection: a machine learning analysis" arXiv:2202.14006