ETM icon indicating copy to clipboard operation
ETM copied to clipboard

Test data partition in evaluation()

Open jeremytanjianle opened this issue 5 years ago • 2 comments

Hi, absolutely love the repo and the paper. Congratulations on developing this.

Would just like to ask about the evaluation function in main.py

Line 275: preds is derived from normalized_data_batch_1 but data_batch_2 is a different set of document altogether.

recon_loss = -(preds * data_batch_2).sum(1) 

What does recon_loss mean if it matches the reconstruction of one set of documents against a different set of documents?

jeremytanjianle avatar Feb 22 '20 02:02 jeremytanjianle

Same question here

yilunzhao avatar Apr 23 '20 08:04 yilunzhao

IMHO data_batch_1 is a batch of first halve of documents and data_batch_2 is a batch of second halve of the same documents, namely document completion task.

kkew3 avatar Dec 29 '23 09:12 kkew3