Add DeepNCMClassifier model

Add tests for DeepNCMClassifier Remove old test Add multi label support Add type hints and doc strings
flairNLP · Nov 24, 2024 · cfeea13 · cfeea13
1 parent 9a962cb
commit cfeea13
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,3 +1,4 @@
+from .deepncm_classification_model import DeepNCMClassifier
 from .entity_linker_model import SpanClassifier
 from .entity_mention_linking import EntityMentionLinker
 from .language_model import LanguageModel
@@ -37,4 +38,5 @@
     "TextClassifier",
     "TextRegressor",
     "MultitaskModel",
+    "DeepNCMClassifier",
 ]