guillitte / pytorch-sentiment-neuron

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Train and Validation file format

loretoparisi opened this issue · comments

Hello, I have created this Docker image for the original sentiment neuron that helps to test prediction of the sentiment feature as well as the generative model. I would like to add training, so I was starting from your Pytorch version.
How is the train and validation dataset format? Did you use the Amazon Reviews Full or Polarity Dataset for training i.e. the kaggle version here?

Thank you very much.