mirror of
https://github.com/ghndrx/kubeflow-pipelines.git
synced 2026-02-10 06:45:13 +00:00
Greg Hendrickson
afc8fc6690
- 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
10 lines
151 B
Plaintext
10 lines
151 B
Plaintext
runpod>=1.7.0
|
|
transformers==4.44.0
|
|
datasets>=2.16.0
|
|
accelerate>=0.30.0
|
|
boto3>=1.34.0
|
|
scikit-learn>=1.3.0
|
|
scipy>=1.11.0
|
|
safetensors>=0.4.0
|
|
PyTDC>=1.1.0
|