PyTDC

When to Use This Skill

This skill should be used when:

• Working with drug discovery or therapeutic ML datasets
• Benchmarking machine learning models on standardized pharmaceutical tasks
• Predicting molecular properties (ADME, toxicity, bioactivity)
• Predicting drug-target or drug-drug interactions
• Generating novel molecules with desired properties
• Accessing curated datasets with proper train/test splits (scaffold, cold-split)
• Using molecular oracles for property optimization

Installation & Setup

Install PyTDC using pip:

To upgrade to the latest version:

Core dependencies (automatically installed):

• numpy, pandas, tqdm, seaborn, scikit_learn, fuzzywuzzy

Additional packages are installed automatically as needed for specific features.

Quick Start

The basic pattern for accessing any TDC dataset follows this structure:

Where:

• <problem>: One of single_pred, multi_pred, or generation
• <Task>: Specific task category (e.g., ADME, DTI, MolGen)
• <Dataset>: Dataset name within that task

Example - Loading ADME data:

from tdc.single_pred import ADME
data = ADME(name='Caco2_Wang')
split = data.get_split(method='scaffold')