pytdc

Load therapeutics drug-discovery datasets and benchmarks via the PyTDC API.

4|Updated Mar 2, 2026
One-click install
npx skills add https://github.com/shushuzn/Rairos --skill pytdc-shushuzn
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: pytdc
Source: https://github.com/shushuzn/Rairos/tree/main/skills/pytdc
Command: npx skills add https://github.com/shushuzn/Rairos --skill pytdc-shushuzn

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) and references (resource) components.

What problem does it solve?

Provides standardized, AI-ready therapeutics datasets and benchmarks so you can evaluate drug discovery and therapeutic ML models with correct splits, consistent formats, and reproducible protocols.

Core Features & Use Cases

  • Curated therapeutic datasets across the pipeline: Supports single-instance prediction (molecular/protein properties), multi-instance prediction (DTI/DDI/PPI-like interactions), and generation tasks (molecule generation and retrosynthesis).
  • Meaningful dataset splits: Enables scaffold splits for chemical diversity and cold splits for DTI/DDI generalization scenarios (unseen drugs and/or targets).
  • Unified evaluation workflow: Works with standardized evaluators and benchmark-group evaluation across multiple seeds for robust comparison.
  • Molecular property oracles for optimization: Provides oracles for property prediction and goal-directed optimization to score candidate molecules.

Quick Start

Use the pytdc skill to load the ADME dataset Caco2_Wang and obtain scaffold-based train/valid/test splits for model evaluation.

Frequently Asked Questions about pytdc

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I load drug discovery datasets for machine learning evaluation?

You can load standardized therapeutics datasets for machine learning evaluation using a unified API that provides task-scoped access to ADME, toxicity, DTI, DDI, and molecular generation benchmarks. This ensures consistent formats and reproducible experimental protocols.

What is a scaffold split and when do I need it for molecular property prediction?

A scaffold split partitions chemical data based on structural diversity, ensuring test molecules have distinct scaffolds from training data. You need it for molecular property prediction to evaluate model generalization on novel chemical structures rather than memorized analogs.

Can I get cold splits for DTI and DDI generalization scenarios?

Yes, you can generate cold splits for DTI and DDI generalization scenarios. The system supports cold-start splits that hold out unseen drugs and/or targets, allowing you to rigorously evaluate interaction prediction models on novel entities.

What's the best way to benchmark molecule optimization workflows?

The best way to benchmark molecule optimization workflows is using built-in molecular property oracles. These oracles score candidate molecules for goal-directed optimization, providing standardized evaluation metrics for property prediction and generation tasks.

Does this therapeutics benchmarking tool support retrosynthesis tasks?

Yes, the therapeutics benchmarking tool supports retrosynthesis tasks. It provides curated datasets and standardized evaluators for molecule generation and retrosynthesis, ensuring AI-ready data formats for machine learning experimentation.

How do I evaluate drug discovery models consistently across multiple seeds?

To evaluate drug discovery models consistently across multiple seeds, apply the standardized evaluator workflow. It enables benchmark-group evaluation across multiple random seeds, yielding robust performance comparisons and highly reproducible results.