test

example/re/document/predict.py

@@ -9,7 +9,7 @@ from torch.utils.data import DataLoader
 from transformers import AutoConfig, AutoModel, AutoTokenizer
 from transformers.optimization import AdamW, get_linear_schedule_with_warmup
 
-from deepke.relation_extraction.document import *
+from deepkeredoc import *
 
 def evaluate(args, model, features, tag="dev"):
 

example/re/document/run.py

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

example/re/few-shot/predict.py

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

setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 setup(
     name='deepke',  # 打包后的包文件名
-    version='0.2.64',    #版本号
+    version='0.2.67',    #版本号
     keywords=["pip", "RE","NER","AE"],    # 关键字
     description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。',  # 说明
     long_description="client",  #详细说明
@@ -14,18 +14,18 @@ setup(
     package_dir={"": "src"},
     packages=find_packages("src"),
     install_requires=[
-        'torch==1.4',
+        'torch==1.5',
         'hydra-core==1.0.6',
-        'tensorboard>=2.0',
-        'matplotlib>=3.1',
-        'tensorboardX>=2.4',
+        'tensorboard==2.4.1',
+        'matplotlib==3.4.1',
+        'tensorboardX==2.4',
         'transformers==3.4.0',
-        'jieba>=0.39',
-        'scikit-learn>=0.22',
-        'pytorch-transformers>=1.2.0',
-        'seqeval>=0.0.5',
-        'tqdm>=4.31.1',
-        'nltk>=3.4.5'
+        'jieba==0.42.1',
+        'scikit-learn==0.24.1',
+        'pytorch-transformers==1.2.0',
+        'seqeval==1.2.2',
+        'tqdm==4.60.0',
+        'nltk==3.6.3'
     ], 
     classifiers=[
         "Programming Language :: Python :: 3",

src/deepke/relation_extraction/__init__.py

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