tabnet-modified icon indicating copy to clipboard operation
tabnet-modified copied to clipboard

Encoder-decoder architecture

Open astrogilda opened this issue 5 years ago • 1 comments

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?

astrogilda avatar Jul 22 '20 23:07 astrogilda

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 image 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.

ptuls avatar Jul 27 '20 03:07 ptuls