sushirank icon indicating copy to clipboard operation
sushirank copied to clipboard

Educational implementation of pointwise and pairwise learning-to-rank models

sushirank

Educational implementation of pointwise and pairwise learning-to-rank models

Slides

Repository for Shopee x Data Science BKK Dive into Learning-to-rank ใครไม่แร้งค์ เลินนิ่งทูแร้งค์

  • sandbox.ipynb - notebook for workshop
  • sushirank/datasets.py - Pytorch datasets for pointwise and pairwise models
  • sushirank/finetuners.py - pointwise and pairwise models implemented in Pytorch-lightning
  • sushirank/metrics.py - common metrics for ranking problems

Resources for Implementation

Research Papers