Commit Graph

4 Commits

Author SHA1 Message Date
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
0f4858d22f fix: disable checkpoint saving to avoid tensor contiguity error 2026-02-03 02:38:15 +00:00
2680ad5502 fix: remove MinIO dependency, use inline training data 2026-02-03 02:27:49 +00:00
9ca3d6c195 Add DDI training pipeline with RunPod serverless GPU support 2026-02-02 23:56:05 +00:00