masterformat-classifier / requirements.txt
constructelligence's picture
Add transformer+TF-IDF ensemble (line-item top-1 0.597 -> 0.686) and document it in the card
9d4c04e verified
Raw History Blame Contribute Delete
206 Bytes
# Core inference (PyTorch)
transformers>=4.40
torch>=2.0
# ONNX path: pip install onnxruntime
# onnxruntime>=1.17
# numpy>=1.24
# Ensemble (ensemble/predict_ensemble.py)
# scikit-learn>=1.3
# joblib>=1.3