Update pythainlp.wsd api

Author: wannaphong

notebooks/test_wsd.ipynb

@@ -14,7 +14,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 2,
    "id": "2ef43b65-5df9-42e3-a712-0e60ca64ea16",
    "metadata": {
     "tags": []
@@ -24,7 +24,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[('โปรแกรมคอมพิวเตอร์ใช้ในทางอินเทอร์เน็ตสำหรับเก็บข้อมูลของผู้ใช้งาน', 0.0974416732788086), ('ชื่อขนมชนิดหนึ่งจำพวกขนมเค้ก แต่ทำเป็นชิ้นเล็ก ๆ แบน ๆ แล้วอบให้กรอบ', 0.09319090843200684)]\n"
+      "Setting ds_accelerator to cuda (auto detect)\n",
+      "[('ชื่อขนมชนิดหนึ่งจำพวกขนมเค้ก แต่ทำเป็นชิ้นเล็ก ๆ แบน ๆ แล้วอบให้กรอบ', 0.09319090843200684), ('โปรแกรมคอมพิวเตอร์ใช้ในทางอินเทอร์เน็ตสำหรับเก็บข้อมูลของผู้ใช้งาน', 0.0974416732788086)]\n"
      ]
     }
    ],
@@ -34,7 +35,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 3,
    "id": "6aafefdf-4658-4e35-b69f-7d2b54e34d33",
    "metadata": {
     "tags": []
@@ -44,7 +45,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[('โปรแกรมคอมพิวเตอร์ใช้ในทางอินเทอร์เน็ตสำหรับเก็บข้อมูลของผู้ใช้งาน', 0.1005704402923584), ('ชื่อขนมชนิดหนึ่งจำพวกขนมเค้ก แต่ทำเป็นชิ้นเล็ก ๆ แบน ๆ แล้วอบให้กรอบ', 0.12473666667938232)]\n"
+      "[('ชื่อขนมชนิดหนึ่งจำพวกขนมเค้ก แต่ทำเป็นชิ้นเล็ก ๆ แบน ๆ แล้วอบให้กรอบ', 0.12473666667938232), ('โปรแกรมคอมพิวเตอร์ใช้ในทางอินเทอร์เน็ตสำหรับเก็บข้อมูลของผู้ใช้งาน', 0.1005704402923584)]\n"
      ]
     }
    ],
@@ -128,7 +129,7 @@
     {
      "data": {
       "text/plain": [
-       "('เดิน', ['ยกเท้าก้าวไป', 'เคลื่อนไปด้วยกำลังต่าง ๆ'])"
+       "('เดิน', ['เคลื่อนไปด้วยกำลังต่าง ๆ', 'ยกเท้าก้าวไป'])"
       ]
      },
      "execution_count": 8,

pythainlp/wsd/core.py

@@ -52,7 +52,7 @@ def get_sense(
     sentence: str,
     word: str,
     device:str="cpu",
-    custom_dict: dict=_mean_all,
+    custom_dict: Union[dict,None]=None,
     custom_tokenizer: Tokenizer=_word_cut,
 ) -> Union[List[Tuple[str, float]], None]:
     """
@@ -96,6 +96,8 @@ def get_sense(
         #   0.12473666667938232)]
     """
     global _MODEL
+    if custom_dict == None:
+        custom_dict = _mean_all
     _w = custom_tokenizer.word_tokenize(sentence)
     if word not in set(custom_dict.keys()) or word not in sentence:
         return None