Skip to content

v0.1.2: Adding WARP Loss

Compare
Choose a tag to compare
@etlundquist etlundquist released this 30 May 18:55
· 88 commits to master since this release
1edef94

Added

  • WARP loss - while slower to train this yields slightly better performance on dense interaction data and much better performance on highly sparse interaction data relative to BPR
  • new hyperparameters loss and max_samples
  • re-wrote the numba _fit() function to elegantly (IMHO) handle both BPR and WARP loss