requirements-train.txt

-r requirements.txt
datasets>=2.16.0
accelerate>=0.25.0
scikit-learn>=1.3.0
pandas>=2.0.0
numpy>=1.24.0
matplotlib>=3.7.0
seaborn>=0.12.0