mljar-supervised icon indicating copy to clipboard operation
mljar-supervised copied to clipboard

Add Tabnet algo

Open bobcolner opened this issue 3 years ago • 1 comments

TabNet: Attentive Interpretable Tabular Learning

Would be great to add this new deep/attention neural net type to mljar. Performance is competitive with boosted trees and should add diversity to ensembles.

https://github.com/dreamquark-ai/tabnet

bobcolner avatar Jul 19 '21 15:07 bobcolner

@bobcolner have you used TabNet? Does it work good?

Maybe this can be added as the extension to mljar-supervised? (not in the main package, but installed additionally)

pplonski avatar Jul 26 '21 09:07 pplonski

Hi @pplonski Interested in contributing adding TabNet into mljar-supervised, may I get some thoughts on your comments "as the extension .... installed additionally" ? what might be an ideal way to add it "installed additionally" ( in my old mind, I am still stuck in the prototype where adding the functionality -> update dependency -> pip install , so I am not sure what would be "installed additionally") Thanks for the hints and inputs!!

pangjac avatar Jan 24 '23 21:01 pangjac

It will be extra package installed by:

pip install mljar-supervised[tabnet]

pplonski avatar Jan 25 '23 06:01 pplonski

I'm closing for now. I don't want to extend required packages, with tabnet it will require pytorch ...

pplonski avatar May 01 '23 13:05 pplonski