ETM
ETM copied to clipboard
Test data partition in evaluation()
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?
Same question here
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.