Binary classification with BCELoss or Focal Loss

uf-hobi-informatics-lab / ClinicalTransformerRelationExtraction

A project for developing transformer-based models for clinical relation extraction

MIT License

125 stars 23 forks source link

Binary classification with BCELoss or Focal Loss #5

Closed bugface closed 3 years ago

bugface commented 3 years ago

For binary mode, we currently still use CrossEntropyLoss, but BCELoss is designed for binary classification. We need to add options to use BCELoss or Focal Loss in binary mode

bugface commented 3 years ago

added