/recommender_algorithms

推薦アルゴリズムで遊んだものの置き場

Primary LanguagePython

Recommender_algorithm

Setup

For get train and test dataset, please run command.

$ ./setup.sh

contents

cpu

  • collaborative filtering
  • matrix factorization
  • factorization machine
  • iALS
  • EASE
  • topic modeling

gpu

  • LightGCN
  • Two-Tower model
  • sequential
  • recsim ng

eval

  • ndcg
  • recall, precision,f1-score

util

  • dataloader
  • data_downloader
  • pipeline