Bert-Pytorch-Chinese-TextClassification
Bert-Pytorch-Chinese-TextClassification copied to clipboard
update for save/load function and some eval indexes
Add 2 args:
- --from_trained(bool default False) To save and load the model you trained
- --is_bi-classification(bool default True) To calculate indexes like precision recall and F1 when doing classification with only 2 labels (like real and fake)