bert_document_classification icon indicating copy to clipboard operation
bert_document_classification copied to clipboard

Threshold value set to 0 in document_bert.py predict

Open zeinramadan opened this issue 4 years ago • 0 comments

The paper claims that the threshold was 0.5 however in the code, specifically in the predict method in document_bert.py, the threshold's default value is 0. In the fit method, the predict function is called with no threshold argument so the threshold is 0. I just thought this could potentially be an error in the code.

zeinramadan avatar Aug 08 '21 01:08 zeinramadan