Change the default NER to thainer-v2

PyThaiNLP · Jun 21, 2023 · 5e97e7c · 5e97e7c
1 parent 61c6b8e
commit 5e97e7c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pythainlp/tag/named_entity.py b/pythainlp/tag/named_entity.py
@@ -27,17 +27,17 @@ class NER:
     :param str corpus: corpus
 
     **Options for engine**
+        * *thainer-v2* - Thai NER engine v2.0 for Thai NER 2.0 (default)
         * *thainer* - Thai NER engine
         * *tltk* - wrapper for `TLTK <https://pypi.org/project/tltk/>`_.
-        * *thainer-v2* - Thai NER engine v2.0 for Thai NER 2.0
 
     **Options for corpus**
-        * *thainer* - Thai NER corpus
+        * *thainer* - Thai NER corpus (default)
 
     **Note**: for tltk engine, It's support ner model from tltk only.
     """
 
-    def __init__(self, engine: str, corpus: str = "thainer") -> None:
+    def __init__(self, engine: str = "thainer-v2", corpus: str = "thainer") -> None:
         self.load_engine(engine=engine, corpus=corpus)
 
     def load_engine(self, engine: str, corpus: str) -> None:

diff --git a/pythainlp/tag/thainer.py b/pythainlp/tag/thainer.py
@@ -25,7 +25,6 @@
 from pythainlp.tokenize import word_tokenize
 from pythainlp.util import isthai
 
-_CORPUS_NAME = "thainer"
 _TOKENIZER_ENGINE = "newmm"  # should be the same as one used in training data