Files
kubeflow-pipelines/components/runpod_trainer/requirements.txt
Greg Hendrickson afc8fc6690 feat: Add real DrugBank DDI dataset support via TDC
- Added PyTDC dependency for DrugBank access
- Implemented DDI type -> severity label mapping (0-4)
- Added train/eval split with stratification
- Added accuracy and F1 metrics for evaluation
- Default: 50K samples from DrugBank DDI
- Supports both real data and custom inline data
2026-02-03 02:48:31 +00:00

151 B