Merge pull request #256 from lsgrep/fix-embedding-config

fix embedding config
BrikerMan · Sep 30, 2019 · 7eddb47 · 7eddb47
2 parents fc7d089 + 6c9c22e
commit 7eddb47
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/kashgari/embeddings/base_embedding.py b/kashgari/embeddings/base_embedding.py
@@ -8,8 +8,8 @@
 # time: 2019-05-20 17:40
 
 import json
-import pydoc
 import logging
+import pydoc
 from typing import Union, List, Optional, Dict
 
 import numpy as np
@@ -33,6 +33,7 @@ def info(self) -> Dict:
             'config': {
                 'sequence_length': self.sequence_length,
                 'embedding_size': self.embedding_size,
+                'task': self.task
             },
             'embed_model': json.loads(self.embed_model.to_json()),
         }

diff --git a/kashgari/processors/classification_processor.py b/kashgari/processors/classification_processor.py
@@ -22,7 +22,7 @@ def __init__(self, multi_label=False, **kwargs):
     def info(self):
         info = super(ClassificationProcessor, self).info()
         info['task'] = kashgari.CLASSIFICATION
-        info['multi_label'] = self.multi_label
+        info['config']['multi_label'] = self.multi_label
         return info
 
     def _build_label_dict(self,

diff --git a/setup.py b/setup.py
@@ -79,4 +79,4 @@ def find_version(*file_paths):
 )
 
 if __name__ == "__main__":
-    print("Hello world")
+    print("Hello world")