tabnet-modified
tabnet-modified copied to clipboard
Encoder-decoder architecture
Hey Paul,
Thanks for this repository! Could you help me understand how one would modify the code to function in an encoder decoder capacity, as referenced in the paper?
Hey @astrogilda,
You'll need to create a new decoder function, similar to the encoder function
def decoder(self, data, reuse, is_training)
You'll have to implement a new loss function as per the paper for self-supervision
in Section 4.
Then, in here, add a decoder after the encoder to train I think. You might want to remove the classification part too.
At least that's the rough idea I have. Hopefully that helps.