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
flairNLP · Nov 24, 2024 · b19e700 · b19e700
1 parent cfeea13
commit b19e700
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 334 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",
 ]