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

WIP chore: minor refacto to simplify code #449

Closed Optimox closed 1 year ago

Optimox commented 1 year ago

What kind of change does this PR introduce?

This is a minor refacto proposed in #447

Does this PR introduce a breaking change?

It does not change anything, just code cleaning

What needs to be documented once your changes are merged?

Closing issues

It does not close the issue since a real change could be implemented to get closer to original paper in pretraining