dreamquark-ai / tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
https://dreamquark-ai.github.io/tabnet/
MIT License
2.55k stars 470 forks source link

fix-438: correct rmsle computation #477

Closed Optimox closed 1 year ago

Optimox commented 1 year ago

What kind of change does this PR introduce?

Minimal bugfix

Does this PR introduce a breaking change?

No What needs to be documented once your changes are merged?

Nothing Closing issues

closes #438