Skip to content

liketheflower/Drug_discovery_project_documentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 

Repository files navigation

Drug_discovery_project_documentation

How to get ready?

Get ready for GNN

GNN Tutorials summary

Get ready to use DGL and Pytorch

Please read the official document of DGL and Pytorch. For advanced knowledge about how to use Pytorch, please read this article. How to understand Pytorch Source Code?

Get ready for the Autoencoder

AE, VAE, and VGAE

Papers we used for this project

This project is mainly based on three papers:

Network structure

We have TWO PARTS:

  1. Unsupervised machine learning part.
  2. Supervised machine learning part.

For the first part, we have two branches, one is used to predict the adjacency matrix as the VGAE paper. One is used to predict the Fingerprint. So, in total, we have three branches as shown here network architecture.

Branch 3: Graph classification tutorial

The graph calssification is based on the GIN. tutorial can be found from Tutorial of Graph Classification by DGL

Branch 1 and 2

Branch 1 and 2 is essential a GAE. It is using the GAE model introducted in the VGAE paper. At the same time, the encoder network is replaced to the network of GIN. Detail of the network structure can be found from Here For the final implementation, the VGINAE branch is based on a variational GINAE implementation. The motiff learning part is using GINAE implementation. The motiff part can also be changed to VGINAE if necessary. The VGINAE's variational implementation based on DGL/Pytorch was modified based on the original tensorflow implentation of the VGAE model. The original tensorflow implementation can be found HERE

Code for the whole system

Code of the whole system can be get from the github. It is a private repo, please sending email to [email protected] to ask for accessing. https://github.com/liketheflower/graph_classification_jak

Whole system design document

Code base overview

How to switch between different branches

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published