neural-collaborative-filtering icon indicating copy to clipboard operation
neural-collaborative-filtering copied to clipboard

Published 20 hours ago • verified publisher icon yihong-chen

Test Dataloader for large dataset.

Open RuihongQiu opened this issue 6 years ago • 2 comments

Maybe a test dataloader for iterate the test set is more user friendly for large dataset? Just simply treat all test data as a huge batch sometimes will cause OOM error.

image

Sometimes the size could be like this.

RuihongQiu avatar Mar 24 '19 11:03 RuihongQiu

Couldn't agree more. A test dataloader would be better. I will add it to my todo list.

yihong-chen avatar Mar 25 '19 02:03 yihong-chen

Hope to see the method. I try to implement it but get stuck in the calculation of HR and NDCG.

RuihongQiu avatar Mar 25 '19 05:03 RuihongQiu