Add predictor for MultiBERT model, improve the trainer for the same model and...
Add predictor for MultiBERT model, improve the trainer for the same model and expose the predicted score in the model's output
Showing
- scripts/ML/BERT/Classifier.py 5 additions, 7 deletionsscripts/ML/BERT/Classifier.py
- scripts/ML/Corpus.py 7 additions, 14 deletionsscripts/ML/Corpus.py
- scripts/ML/predictMulti.py 26 additions, 0 deletionsscripts/ML/predictMulti.py
- scripts/ML/predictSimple.py 1 addition, 1 deletionscripts/ML/predictSimple.py
- scripts/ML/prodigy-multi-jsonl-to-tsv.py 1 addition, 1 deletionscripts/ML/prodigy-multi-jsonl-to-tsv.py
- scripts/ML/trainMultiBERT.py 7 additions, 13 deletionsscripts/ML/trainMultiBERT.py
Loading
Please register or sign in to comment