What problem does it solve?
TorchDrug removes the complexity of building graph-based machine learning workflows for chemistry and biology, helping you predict properties, reason over knowledge graphs, generate molecules, and plan syntheses from a unified toolkit.
Core Features & Use Cases
- Molecular property prediction: Build classifiers and regressors for ADMET, toxicity, solubility, and quantum chemistry benchmarks.
- Protein modeling: Work with sequence or structure data to predict enzyme function, localization, stability, contacts, and protein interactions.
- Knowledge graph reasoning and retrosynthesis: Perform link prediction on biomedical graphs like Hetionet and decompose synthesis planning into reaction-center identification and synthon completion.
- Use case: A researcher can start with BBBP, EnzymeCommission, or Hetionet and quickly choose a strong baseline model, dataset split, and metric set for a publishable experiment.
Quick Start
Use the torchdrug skill to set up a graph learning workflow for your dataset, select the recommended model and task, and evaluate it with the appropriate metrics and split strategy.