Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 

README.md

MoHR: Recommendation Through Mixtures of Heterogeneous Item Relationships

This is our TensorFlow implementation for the paper:

Wang-Cheng Kang, Mengting Wan, Julian McAuley (2018). Recommendation Through Mixtures of Heterogeneous Item Relationships. In Proceedings of ACM Conference on Information and Knowledge Management (CIKM'18)

Please cite our paper if you use the code or datasets.

The code is tested under a Linux desktop with TensorFlow 1.12.

Datasets

The Automotive from Amazon is included in the repo. All datasets (after pre-processing) can be downloaded from:

Model Training

A simple way to train our model is (with default hyper-parameters):

python main.py --dataset=Automotive 

For the Automotive dataset, the model should be converged in 600 epochs, you should be able to see the test AUC in the log file reach 0.8.

For more details (e.g. learning rate, regularizations, etc), please refer to the code.

About

MoHR: Recommendation Through Mixtures of Heterogeneous Item Relationships

Topics

Resources

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.