ULTR-Community / ULTRA_pytorch

Unbiased Learning To Rank Algorithms (ULTRA)
https://ultr-community.github.io/ULTRA_pytorch/
Apache License 2.0
93 stars 8 forks source link

Fix bugs in Pytorch Vesion #7

Closed rowedenny closed 2 years ago

rowedenny commented 2 years ago
  1. Fix loss with stacking memory during the model training
  2. Fix the ERR metric computation
  3. Fix the evaluation shallow copy for dict summary