Skip to content
/ CF-UIcA Public

Code for "Collaborative Filtering with User-Item Co-Autoregressive Models"

Notifications You must be signed in to change notification settings

thu-ml/CF-UIcA

Repository files navigation

CF-UIcA

This repository contains the implementation for the paper:

Chao Du, Chongxuan Li, Yin Zheng, Jun Zhu, Bo Zhang. Collaborative Filtering with User-Item Co-Autoregressive Models. In AAAI 2018.

Datasets

To download and preprocess the Movielens 1M dataset:

$ cd data/
$ ./download.sh
$ python preprocess.py
$ python preprocess_implicit.py

Usage

To run rating prediction experiments:

$ python CF-UIcA_rating-prediction.py

To run top-N recommendation experiments:

$ python CF-UIcA_topN-recommendation.py

Reference

Please cite our AAAI'18 paper if you find it is useful. Thanks!

@inproceedings{du2018collaborative,
    title={Collaborative Filtering with User-Item Co-Autoregressive Models},
    author={Chao Du and Chongxuan Li and Yin Zheng and Jun Zhu and Bo Zhang},
    booktitle={AAAI},
    year={2018}
}

About

Code for "Collaborative Filtering with User-Item Co-Autoregressive Models"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published