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-478: add n_shared_decoder and n_indep_decoder #479

Closed Optimox closed 1 year ago

Optimox commented 1 year ago

What kind of change does this PR introduce?

This fixes the bug #478, it adds back the two missing parameters

Does this PR introduce a breaking change?

No breaking change.

What needs to be documented once your changes are merged? No need for documentation as it's just a bug fix Closing issues closes #478