pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
https://pytorch-ignite.ai
BSD 3-Clause "New" or "Revised" License
4.51k stars 609 forks source link

Implementation of RecSys Metrics #2631

Open kamalojasv181 opened 2 years ago

kamalojasv181 commented 2 years ago

🚀 Feature

To add some more metrics used in Recommendation Systems with tests that can directly be used. Here is the list (@alykhantejani @vfdev-5 feel free to add more):

shivance commented 1 year ago

Can I contribute to this one ?

vfdev-5 commented 1 year ago

@shivance yes, sure. Do you have a recsys metric in mind to implement ?

shivance commented 1 year ago

FactorizedTopK could be a place to start? I've also seen some systems implement Mean Average Precision MAP

I'm not aware of many metrics specific to recsys, so am open to learn !

vfdev-5 commented 1 year ago

@shivance how about implementing hitrate, mean reciprocal rank etc ? Some refs to check: