Skip to content

The official implementation of "Disentangling User Interest and Conformity for Recommendation with Causal Embedding" (WWW '21)

License

Notifications You must be signed in to change notification settings

tsinghua-fib-lab/DICE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DICE

This is the official implementation of our WWW'21 paper:

Yu Zheng, Chen Gao, Xiang Li, Xiangnan He, Depeng Jin, Yong Li, Disentangling User Interest and Conformity for Recommendation with Causal Embedding, In Proceedings of the Web Conference 2021.

Model training

First unzip the datasets and start the visdom server:

visdom -port 33336

Then simply run the following command to reproduce the experiments on corresponding dataset and model:

python app.py --flagfile ./config/xxx.cfg

Embedding visualization

The visualization codes to reproduce Figure 5(b) and Figure 7 can be found in the viz folder.

First, reduce the dimension of the embedding vectors to 2D using t-SNE (remember to change the path to the model checkpoint in viz.py):

python viz.py

Then, visualize the 2D embedding vectors using MATLAB:

embedding_viz.m

Dataset processing

The dataset process codes are in this repo. Please check this issue for more details.

Citation

If you use our codes and datasets in your research, please cite:

@inproceedings{zheng2021disentangling,
  title={Disentangling User Interest and Conformity for Recommendation with Causal Embedding},
  author={Zheng, Yu and Gao, Chen and Li, Xiang and He, Xiangnan and Li, Yong and Jin, Depeng},
  booktitle={Proceedings of the Web Conference 2021},
  pages={2980--2991},
  year={2021}
}

About

The official implementation of "Disentangling User Interest and Conformity for Recommendation with Causal Embedding" (WWW '21)

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published