Neural Collaborative Reasoning

This repository includes the implementation for Neural Collaborative Reasoning (NCR):

Hanxiong Chen, Shaoyun Shi, Yunqi Li, Yongfeng Zhang. 2021. Neural Collaborative Reasoning. In Proceedings of the Web Conference 2021(WWW ’21)

Refernece

For inquiries contact Hanxiong Chen ([email protected]) or Yongfeng Zhang ([email protected])

@inproceedings{chen2021neural,
  title={Neural Collaborative Reasoning},
  author={Hanxiong Chen, Shaoyun Shi, Yunqi Li, and Yongfeng Zhang},
  booktitle={Proceedings of the the Web Conference 2021},
  year={2021}
}

Environments

Python 3.6.6

Packages: See in requirements.txt

numpy==1.18.1
torch==1.0.1
pandas==0.24.2
scipy==1.3.0
tqdm==4.32.1
scikit_learn==0.23.1

Datasets

• The processed datasets are in ./dataset/

• ML-100k: The origin dataset can be found here.

• Amazon Datasets: The origin dataset can be found here.

Example to run the codes

• To guarantee the program can execute properly, please keep the directory structure as given in this repository.
• Some running commands can be found in ./command/command.py
• For example:

# Neural Collaborative Reasong on ML-100k dataset
> cd NCR/src/
> python main.py --rank 1 --model_name NCR --optimizer Adam --lr 0.001 --dataset ml100k01-1-5 --metric ndcg@5,ndcg@10,hit@5,hit@10 --max_his 5 --test_neg_n 100 --l2 1e-4 --r_weight 0.1 --random_seed 2022 --gpu 0