dreamquark-ai / tabnet

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

fix: correct rmsle loss computation #448

Closed Optimox closed 1 year ago

Optimox commented 1 year ago

What kind of change does this PR introduce?

Minor fix from #438

Does this PR introduce a breaking change?

No What needs to be documented once your changes are merged? Nothing, just a bug fix

Closing issues closes #438