Add DeepNCMClassifier model

Add tests for DeepNCMClassifier Remove old test Add multi label support Add type hints and doc strings
flairNLP · Nov 23, 2024 · 224697a · 224697a
1 parent 091e62b
commit 224697a
Show file tree

Hide file tree

Showing 5 changed files with 667 additions and 0 deletions.
diff --git a/flair/models/__init__.py b/flair/models/__init__.py
@@ -1,4 +1,5 @@
 from .clustering import ClusteringModel
+from .deepncm_classification_model import DeepNCMClassifier
 from .entity_linker_model import SpanClassifier
 from .entity_mention_linking import EntityMentionLinker
 from .language_model import LanguageModel
@@ -39,4 +40,5 @@
     "TextRegressor",
     "ClusteringModel",
     "MultitaskModel",
+    "DeepNCMClassifier",
 ]