search

graphnet-team / graphnet

A Deep learning library for neutrino telescopes
https://graphnet-team.github.io/graphnet/
Apache License 2.0
90 stars 92 forks source link

RMSE von Mises-Fisher 3d loss #716

Closed Aske-Rosted closed 4 months ago

Aske-Rosted commented 4 months ago

Adds a combination of the RMSE loss and the von Mises-Fisher 3d loss, with the Mises-Fisher weighted with a factor 0.05. Similar to what was used in the 2nd place solution in the kaggle competition.