Skip to content

New Classification Models

Latest
Compare
Choose a tag to compare
@ThilinaRajapakse ThilinaRajapakse released this 01 Feb 20:38
· 550 commits to master since this release

New Classification Models

Added

  • Added class weights support for Longformer classification
  • Added new classification models (multilabel classification is not supported yet):
    • DeBERTa
    • MPNet
    • SqueezeBert (no sliding window support)

Changed

  • Updated ClassificationModel logic to make it easier to add new models