https://github.com/ahxt/neuacf
Implementation of paper Aspect-Level Deep Collaborative Filtering via Heterogeneous Information Networks
https://github.com/ahxt/neuacf
Last synced: 11 months ago
JSON representation
Implementation of paper Aspect-Level Deep Collaborative Filtering via Heterogeneous Information Networks
- Host: GitHub
- URL: https://github.com/ahxt/neuacf
- Owner: ahxt
- Created: 2018-07-13T11:58:18.000Z (about 8 years ago)
- Default Branch: master
- Last Pushed: 2019-06-05T17:12:09.000Z (about 7 years ago)
- Last Synced: 2025-04-06T10:25:48.972Z (over 1 year ago)
- Language: Python
- Homepage:
- Size: 73.3 MB
- Stars: 32
- Watchers: 1
- Forks: 7
- Open Issues: 8
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# NeuACF
This is an implementation of paper [(Aspect-Level Deep Collaborative Filtering via Heterogeneous Information Networks)](https://www.ijcai.org/proceedings/2018/0471.pdf).
Please refer our paper if you use this code and the bibtex of this paper is:
```
@inproceedings{han2018aspect,
title={Aspect-Level Deep Collaborative Filtering via Heterogeneous Information Networks.},
author="Han, Xiaotian and Shi, Chuan and Wang, Senzhang and Philip, S Yu and Song, Li",
booktitle={IJCAI},
pages={3393--3399},
year={2018}
}
```
### Requirements
- Python 3.6
- Tensorflow 1.2.1
- docopt 0.6.2
- numpy 1.13.3
- sklearn 0.18.1
- pandas 0.20.1
- scipy 1.0.0
### How to Run
1. unzip dataset.7z
2. Compute the aspect-level similarity matrix with the matlab code
3. Run the model with the python code acf.py
example:
```
python ./acf.py ../dataset/amazon/ amovie --mat "U.UIU,I.IUI,U.UICIU,I.ICI" --epochs 40 --last_layer_size 64 --batch_size 1024 --num_of_neg 10 --learn_rate 0.00005 --num_of_layers 2 --mat_select median
```
### Parameters
Parameter | Note
|:---|:---|
|--mat|sim_mat [default: ""]|
|--epochs|Embedding size [default: 40]|
|--last_layer_size| The number of iterations [default: 64]|
|--num_of_layers| The number of layers [default: 2]|
|--num_of_neg| The number of negs [default: 2]|
|--learn_rate| The learn_rate [default: 0.00005]|
|--batch_size| batch_size [default: 1024]|
|--mat_select| mat select type [default: median]|
|--merge| batch_size [default: attention]|
#### Link
For more information, visit the webpage http://www.shichuan.org