Skip to content

Latest commit

 

History

History
73 lines (44 loc) · 2 KB

README.md

File metadata and controls

73 lines (44 loc) · 2 KB
Raw

AutoDebias

This is the official pytorch implementation of AutoDebias, a debiasing method for recommendation system. AutoDebias is proposed in the paper:

AutoDebias: Learning to Debias for Recommendation

by Jiawei Chen, Hande Dong, Yang Qiu, Xiangnan He, Xin Xin, Liang Chen, Guli Lin and Keping Yang

Published at SIGIR 2021.

Introduction

AutoDebias is an automatic debiasing method for recommendation system based on meta learning, exploiting a small amout of uniform data to learn de-biasing parameters and using these parameters to guide the learning of the recommendation model.

Environment Requirement

The code runs well under python 3.8.5. The required packages are as follows:

  • pytorch == 1.4.0
  • numpy == 1.19.1
  • scipy == 1.5.2
  • pandas == 1.1.3
  • cppimport == 20.8.4.2

Datasets

We use two public datasets (Yahoo!R3 and Coat) and a synthetic dataset (Simulation).

  • user.txt: biased data collected by normal policy of recommendation platform. For Yahoo!R3 and Coat, each line is user ID, item ID, rating of the user to the item. For Simulation, each line is user ID, item ID, position of the item, binary rating of the user to the item.
  • random.txt: unbiased data collected by stochastic policy where items are assigned to users randomly. Each line in the file is user ID, item ID, rating of the user to the item.

Run the Code

Explicit feedback

  • For dataset Yahoo!R3:
python train_explicit.py --dataset yahooR3
  • For dataset Coat:
python train_explicit.py --dataset coat

Implicit feedback

  • For dataset Yahoo!R3:
python train_implicit.py --dataset yahooR3
  • For dataset Coat:
python train_implicit.py --dataset coat

Feedback on list recommendation

  • For dataset Simulation:
python train_list.py --dataset simulation

Contact

Please contact [email protected] or [email protected] if you have any questions about the code and paper.