Skip to content

Modifying SE(3)-equivariant GNN to work on custom dataset for the 2024 NeurIPS Kaggle competition.

Notifications You must be signed in to change notification settings

ndriggs/DeepLearningExamples

 
 

Repository files navigation

SE(3) Equivariant GNN

Forked NVIDIA's code for SE(3) Equivariant GNN to modify it for molecular property (specific protein binding affinity) prediction on a custom dataset for the NeurIPS 2024 Leash Bio Kaggle Competition. My code and modifications are in the leash-bio branch.

About

Modifying SE(3)-equivariant GNN to work on custom dataset for the 2024 NeurIPS Kaggle competition.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 50.0%
  • Python 43.3%
  • Shell 2.8%
  • C++ 2.3%
  • Cuda 1.1%
  • Makefile 0.2%
  • Other 0.3%