Contain the drug-disease interactions dataset.
Contain the PPMI matrices of ten drug-related networks.
deep_network_fusion
directory, which contains 3 variants implementations of Multimodal Deep Autoencoder (MDA)recommendation
directory, which contains implementation of Collective Variational Autoencoder (cVAE)- Updated for calculating Recall@K metric.
- To get drug features learned by MDA, run
python get_features.py params.txt
- To predict drug-disease associations by cVAE, run
- Pretraining with features:
python models/recommendation/collective_variational_autoencoder.py --dir dataset -a 6 -b 0.1 -m 300 --save --layer 1000 100
- Refine training with rating:
python models/recommendation/collective_variational_autoencoder.py --dir dataset --rating -a 15 -b 3 -m 500 --load 1 --layer 1000 100
- Pretraining with features:
- On colab, simply follow the
deepDR.ipynb
notebook.