Skip to content
An implementation of RaFM. Xiaoshuang Chen, Yin Zheng, Jiaxiang Wang, et al. "RaFM: Rank-Aware Factorization Machines"
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
buckets/frappe_l
LoadData.py
README.md
RaFM.py
common.py RaFM submitted May 11, 2019
params.py
statistics.py

README.md

RaFM: Rank-Aware Factorization Machines

This is an implementation for the following paper:

Xiaoshuang Chen, Yin Zheng, Jiaxing Wang, Wenye Ma, Junzhou Huang. RaFM: Rank-Aware Factorization Machines. ICML 2019.

Please cite our ICML 2019 paper if you find the paper and the codes beneficial for your research. Thanks!

Dependency

The code is based on Tensorflow, and we have tested it on Python 3.6, Tensorflow 1.13.

Usage

Use statistics.py to compute the frequencies of the features occuring in the datasets.

Use RaFM.py to run the RaFM model.

Default parameters are provided, so you can execute them directly. You can also set it by yourself (see parse_args in each file).

You can’t perform that action at this time.