Variational Graph Auto-Encoders with matrix factorization

This is a TensorFlow implementation of the Variational Graph Auto-Encoder with matrix factorization model for miRNA-disease associations prediction as described in our paper:

“Predicting miRNA-Disease Associations Based on Multi-View Variational Graph Auto-Encoder with Matrix Factorization”

By Yulian Ding, Xiujuan Lei, Bo Liao, Fang-Xiang Wu

Installation

python setup.py install

Requirements

• TensorFlow (1.0 or later)
• python 2.7
• networkx
• scikit-learn
• scipy

Run the demo

python train.py

Data

We provide two kinds of database includes HMDD 2.0 and HMDD 3.2

Cite

Please cite our paper if you use this code in your own work:

“Predicting miRNA-Disease Associations Based on Multi-View Variational Graph Auto-Encoder with Matrix Factorization”