minsoo9506 / RecModel

추천 알고리즘 구현

• 추천알고리즘을 구현하는 레포지토리입니다.
• 라이브러리화 하지는 않습니다.
• '구현을 통해 모델 이해하기', '다른 코드 읽어보기'를 목표로 합니다.

Algorithm

Factorization Machines

• Factorization Machines, 2010
• code: torch model, KMRD-small example

NeuMF

• Neural Collaborative Filtering, 2017
• code: torch model, lit model, KMRD-small example

DeepFM

• DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, 2017
• code: torch model, lit model, KMRD-small example

Two-Tower

• Sampling-Bias-Corrected Neural Modeling for Large Corpus Item Recommendations, RecSys 2019
• code: torch model, lit model, KuaiRec example (torch)
• code: tf model, KuaiRec example (tf)

Benchmark

Data

• 데이터에 대한 설명, 예시데이터 추가

Result

• 모델을 돌린 metric 결과, 훈련시간 등