Where is the topic vector based on VAE

something678 / TodKat

Transformer encoder-decoder for emotion detection in dialogues

MIT License

57 stars 12 forks source link

Where is the topic vector based on VAE #16

Open LeMei opened 1 year ago

LeMei commented 1 year ago

I can not find the topic vector code based on VAE. Would someone find it?

MrZilinXiao commented 1 year ago

I believe it is located here: https://github.com/something678/TodKat/blob/1f711496459c521796043f07861a259c6fe37df5/src/roberta_with_finetune.py#L191

However, no trace of topic pretaining VAE code was found.

MrZilinXiao commented 1 year ago

@LeMei Sorry that I made a mistake. get_topic_vec is not the topic encoder mentioned in the paper. I believe the code is missing crucial parts of the paper's contributions.