EvelynXuNU's picture
Deploy Multi-LexSum demo
ce7c1f0 verified
Raw
History Blame Contribute Delete
409 Bytes
"""Classification subpackage (§3).
Public modules:
- data.py : unified data loader for both tasks
- classical.py : Naive Bayes + Logistic Regression (W7)
- lstm.py : Bi-LSTM with self-trained Word2Vec init (W8)
- bert.py : Legal-BERT fine-tuning (W8)
- train.py : unified CLI training entry point
- explain.py : SHAP-based explanations for the linear models
"""