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 · c92f501 · c92f501
1 parent 224697a
commit c92f501
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 335 deletions.
diff --git a/flair/models/__init__.py b/flair/models/__init__.py
@@ -1,5 +1,4 @@
-from .clustering import ClusteringModel
-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
@@ -40,5 +39,5 @@
     "TextRegressor",
     "ClusteringModel",
     "MultitaskModel",
-    "DeepNCMClassifier",
+    "DeepNCMDecoder",
 ]