COINS icon indicating copy to clipboard operation
COINS copied to clipboard

Questions about data formatting

Open id4thomas opened this issue 2 years ago • 2 comments

Hi, I am currently trying to reproduce your work (specifically COINS GR) and have a few questions about the training data.

From your paper it seems the training data for Knowledge Model would be

  • [SOS] S1 S2 [SEP] S5 [EOS] S2 # EFFECT # Effect2
  • [SOS] S1 S2 [SEP] S5 [EOS] S5 # CAUSE # Casue2
  • [SOS] S1 S2 S3 [SEP] S5 [EOS] S3 # EFFECT # Effect3
  • [SOS] S1 S2 S3 [SEP] S5 [EOS] S5 # CAUSE # Cause3

and

Story Model would be

  • Cause2 [SEP] Effect2 [EOK] [SOS] S1 S2 [SEP] S5 [EOS] S3
  • Cause3 [SEP] Effect3 [EOK] [SOS] S1 S2 S3 [SEP] S5 [EOS] S3

But looking at the part where you load the data (https://github.com/Heidelberg-NLP/COINS/blob/main/model/src/data/conceptnet.py) it is confusing which corresponds to which. Also, the data downloaded with the given script doens't match the format used in the rest of the code

It would be nice if you could provide a data sample for each Knowledge and Story Models or the model weight if possible.

Thank you

id4thomas avatar Jun 16 '22 09:06 id4thomas