Skip to content

hauldhut/GraOmicDRP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Resources

Source code

Regression model

  • Single -omic data
    • GE
    • MUT
    • METH
  • Multiple -omic data
    • GE_MUT_METH
    • GEN_MUT
    • GE_METH
    • MUT_METH

Classification model (To compare with MOLI method)

  • Multiple -omic data
    • GE_MUT_METH
    • GEN_MUT
    • GE_METH
    • MUT_METH

Dependencies

  • Torch
  • Pytorch_geometric
  • Rdkit
  • Matplotlib
  • Pandas
  • Numpy
  • Scipy

Step-by-step running

Regression model:

Create data

python preprocess.py --choice 0

choice: 0: create mixed test dataset 1: create saliency map dataset 2: create blind drug dataset 3: create blind cell dataset

This returns file pytorch format (.pt) stored at data/processed including training, validation, test set.

Train model

python training.py --model 0 --train_batch 1024 --val_batch 1024 --test_batch 1024 --lr 0.0001 --num_epoch 300 --log_interval 20 --cuda_name "cuda:0"

model: 1: GINConvNet 2: GATNet 3: GAT_GCN 4: GCNNet

To train a model using training data. The model is chosen if it gains the best MSE for testing data.

This returns the model and result files for the modelling achieving the best MSE for testing data throughout the training.2

Classification model:

Create data

python preprocess.py --choice 0

Train model

  • Need to run Regession model first to get the weight for the feature extraction task
  • Run end to end jupyter notebook file (remember change the correct data path)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published