Skip to content

Latest commit

 

History

19 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 

Repository files navigation

ACtriplet

ACtriplet: an improved deep learning model for activity cliffs prediction by integrating triplet loss and pre-training

Requirements

  • python = 3.8
  • pytorch = 2.0
  • torch geometric
  • rdkit
  • numpy
  • pandas
  • prefetch-generator
  • pyyaml
  • moleculeACE

Pre-train

Step 1: Prepare dataset

Extract Ac and non-Ac from the 30 macromolecular targets dataset
data pre/get_collection.ipynb

Step 2: Pre-train

python pretrain/main pretrain.py

Finetune

python finetune/main finetune.py

About

No description, website, or topics provided.

Resources

Stars

3 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages