NYXFLOWER / DeCA

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DeCA

This is an implementation for our WWW 2022 paper Learning Robust Recommenders through Cross-Model Agreement.

Requirements

  • torch == 1.9.0+cu102
  • Numpy
  • python3
  • scipy
  • pandas

Datasets

Parameters

Key parameters are all provided in the file configs.py, and you can let the code choose the specific parameters for the model and the dataset with "python xxx.py --default".

Commands

We provide following commands for our methods DeCA and DeCA(p). Simply run the code below will return the results shown in the paper:

python main.py --model GMF --dataset ml-100k --method DeCA --default

where --default means using the default setting. --model is the model drawn from GMF, NeuMF, CDAE, LightGCN, --dataset should be in ml-100k, modcloth, adressa, electronics, --method need to be in DeCA, DeCAp. Remove the --method term, the code will run normal training.
If you want to use your own settings, try:

python main.py --model GMF --dataset modcloth --C_1 1000 --C_2 10 --alpha 0.5 --method DeCA

Citation

If you use our codes in your research, please cite our paper.

About


Languages

Language:Python 100.0%