/tools/3DInfomax

3DInfomax

HannesStark/3DInfomax

172 stars33 forksPythonAdded February 5, 2026
summary

3DInfomax enhances graph neural networks for predicting molecular properties by leveraging 3D molecular geometry. It provides tools for pre-training and fine-tuning models on various molecular datasets, enabling better predictions and molecular fingerprint generation.

description

Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.

topics

graph-representation-learningself-supervised-learninggraphmolecules3dpytorch-geometricdgl-graphgnngraph-neural-networks

Ratings

N/A
0 ratings

Rate this tool:

Comments (0)

Sign in to leave a comment.

No comments yet. Be the first to share your thoughts!