feat: change DeepNCM classifier to a decoder so it can be used with d…

…ifferent model types. make small changes to DefaultClassifier forward_loss to pass label tensor when needed. update tests
flairNLP · Nov 24, 2024 · f94a56f · f94a56f
1 parent cfeea13
commit f94a56f
Show file tree

Hide file tree

Showing 5 changed files with 103 additions and 314 deletions.
diff --git a/flair/models/__init__.py b/flair/models/__init__.py
@@ -1,4 +1,4 @@
-from .deepncm_classification_model import DeepNCMClassifier
+from .deepncm_classification_model import DeepNCMDecoder
 from .entity_linker_model import SpanClassifier
 from .entity_mention_linking import EntityMentionLinker
 from .language_model import LanguageModel
@@ -38,5 +38,5 @@
     "TextClassifier",
     "TextRegressor",
     "MultitaskModel",
-    "DeepNCMClassifier",
+    "DeepNCMDecoder",
 ]