6 lines (5 loc) · 554 Bytes

Ifm-PyTorch

A PyTorch package to build recommendation algorithms for implicit feedback. It provides implementation for WARP and k-order statistic ranking losses in PyTorch.