Bert-Pytorch-Chinese-TextClassification icon indicating copy to clipboard operation
Bert-Pytorch-Chinese-TextClassification copied to clipboard

update for save/load function and some eval indexes

Open 807544076 opened this issue 10 months ago • 0 comments

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)

807544076 avatar Apr 07 '24 09:04 807544076