test

example/re/document/predict.py

@@ -10,7 +10,7 @@ from torch.utils.data import DataLoader
 from transformers import AutoConfig, AutoModel, AutoTokenizer
 from transformers.optimization import AdamW, get_linear_schedule_with_warmup
 
-from deepkeredoc import *
+from deepke.relation_extraction.document import *
 
 
 def report(args, model, features):

example/re/document/run.py

@@ -10,7 +10,7 @@ from torch.utils.data import DataLoader
 from transformers import AutoConfig, AutoModel, AutoTokenizer
 from transformers.optimization import AdamW, get_linear_schedule_with_warmup
 
-from deepkeredoc import *
+from deepke.relation_extraction.document import *
 
 
 def train(args, model, train_features, dev_features, test_features):

example/re/few-shot/predict.py

@@ -13,7 +13,7 @@ from transformers.optimization import get_linear_schedule_with_warmup
 import os
 from tqdm import tqdm
 
-from deepkerefew import *
+from deepke.relation_extraction.few_shot import *
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 

example/re/few-shot/run.py

@@ -13,7 +13,7 @@ from transformers.optimization import get_linear_schedule_with_warmup
 import os
 from tqdm import tqdm
 
-from deepkerefew import *
+from deepke.relation_extraction.few_shot import *
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 

setup.py

@@ -1,11 +1,11 @@
 from setuptools import setup, find_packages
 setup(
     name='deepke',  # 打包后的包文件名
-    version='0.2.79',    #版本号
+    version='0.2.82',    #版本号
     keywords=["pip", "RE","NER","AE"],    # 关键字
     description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。',  # 说明
     long_description="client",  #详细说明
-    license="Apache-2.0 License",  # 许可
+    license="MIT",  # 许可
     url='https://github.com/zjunlp/deepke',
     author='ZJUNLP',
     author_email='xx2020@zju.edu.cn',
@@ -25,7 +25,9 @@ setup(
         'pytorch-transformers==1.2.0',
         'seqeval==1.2.2',
         'tqdm==4.60.0',
-        'nltk==3.6.3'
+        'nltk==3.6.3',
+        'opt-einsum==3.3.0',
+        "ujson"
     ], 
     classifiers=[
         "Programming Language :: Python :: 3",

src/deepke/relation_extraction/__init__.py

@@ -1 +1,3 @@
 from .standard import *
+from .document import *
+from .few_shot import *

src/deepke/relation_extraction/few-shot/lit_models/__init__.py

@@ -1,2 +0,0 @@
-from .transformer import *
-from .base import BaseLitModel 

src/deepke/relation_extraction/few_shot/lit_models/__init__.py

@@ -0,0 +1,3 @@
+from .transformer import *
+from .base import BaseLitModel 
+from .util import *