From 115f6ca852375f5c28451ab82fc6aa8d073a97d6 Mon Sep 17 00:00:00 2001 From: tlk-dsg <467460833@qq.com> Date: Tue, 28 Sep 2021 16:59:43 +0800 Subject: [PATCH] test --- setup.py | 2 +- .../attribution_extraction/standard/models/BiLSTM.py | 5 ++++- .../attribution_extraction/standard/models/Capsule.py | 9 ++++++--- src/deepke/attribution_extraction/standard/models/GCN.py | 9 ++++++--- src/deepke/attribution_extraction/standard/models/LM.py | 7 +++++-- .../attribution_extraction/standard/models/PCNN.py | 7 +++++-- .../standard/models/Transformer.py | 9 ++++++--- src/deepke/relation_extraction/standard/models/BiLSTM.py | 5 ++++- .../relation_extraction/standard/models/Capsule.py | 9 ++++++--- src/deepke/relation_extraction/standard/models/GCN.py | 9 ++++++--- src/deepke/relation_extraction/standard/models/LM.py | 7 +++++-- src/deepke/relation_extraction/standard/models/PCNN.py | 7 +++++-- .../relation_extraction/standard/models/Transformer.py | 9 ++++++--- 13 files changed, 65 insertions(+), 29 deletions(-) diff --git a/setup.py b/setup.py index 190f1c4..641f543 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ from setuptools import setup, find_packages setup( name='deepke', # 打包后的包文件名 - version='0.2.27', #版本号 + version='0.2.40', #版本号 keywords=["pip", "RE","NER","AE"], # 关键字 description='DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。', # 说明 long_description="client", #详细说明 diff --git a/src/deepke/attribution_extraction/standard/models/BiLSTM.py b/src/deepke/attribution_extraction/standard/models/BiLSTM.py index 91cb61d..57803c9 100644 --- a/src/deepke/attribution_extraction/standard/models/BiLSTM.py +++ b/src/deepke/attribution_extraction/standard/models/BiLSTM.py @@ -1,6 +1,9 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch.nn as nn from . import BasicModule -from ..module import Embedding, RNN +from module import Embedding, RNN class BiLSTM(BasicModule): diff --git a/src/deepke/attribution_extraction/standard/models/Capsule.py b/src/deepke/attribution_extraction/standard/models/Capsule.py index 04c9edf..e5d13dd 100644 --- a/src/deepke/attribution_extraction/standard/models/Capsule.py +++ b/src/deepke/attribution_extraction/standard/models/Capsule.py @@ -1,9 +1,12 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch from . import BasicModule -from ..module import Embedding, CNN -from ..module import Capsule as CapsuleLayer +from module import Embedding, CNN +from module import Capsule as CapsuleLayer -from ..utils import seq_len_to_mask, to_one_hot +from utils import seq_len_to_mask, to_one_hot class Capsule(BasicModule): diff --git a/src/deepke/attribution_extraction/standard/models/GCN.py b/src/deepke/attribution_extraction/standard/models/GCN.py index 4df7d40..703e437 100644 --- a/src/deepke/attribution_extraction/standard/models/GCN.py +++ b/src/deepke/attribution_extraction/standard/models/GCN.py @@ -1,10 +1,13 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch import torch.nn as nn from . import BasicModule -from ..module import Embedding -from ..module import GCN as GCNBlock +from module import Embedding +from module import GCN as GCNBlock -from ..utils import seq_len_to_mask +from utils import seq_len_to_mask class GCN(BasicModule): diff --git a/src/deepke/attribution_extraction/standard/models/LM.py b/src/deepke/attribution_extraction/standard/models/LM.py index 25ebe55..91e9cab 100644 --- a/src/deepke/attribution_extraction/standard/models/LM.py +++ b/src/deepke/attribution_extraction/standard/models/LM.py @@ -1,9 +1,12 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) from torch import nn from . import BasicModule -from ..module import RNN +from module import RNN from transformers import BertModel -from ..utils import seq_len_to_mask +from utils import seq_len_to_mask class LM(BasicModule): diff --git a/src/deepke/attribution_extraction/standard/models/PCNN.py b/src/deepke/attribution_extraction/standard/models/PCNN.py index fac9275..b492782 100644 --- a/src/deepke/attribution_extraction/standard/models/PCNN.py +++ b/src/deepke/attribution_extraction/standard/models/PCNN.py @@ -1,10 +1,13 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch import torch.nn as nn import torch.nn.functional as F from . import BasicModule -from ..module import Embedding, CNN +from module import Embedding, CNN -from ..utils import seq_len_to_mask +from utils import seq_len_to_mask class PCNN(BasicModule): diff --git a/src/deepke/attribution_extraction/standard/models/Transformer.py b/src/deepke/attribution_extraction/standard/models/Transformer.py index 44badb6..872e87f 100644 --- a/src/deepke/attribution_extraction/standard/models/Transformer.py +++ b/src/deepke/attribution_extraction/standard/models/Transformer.py @@ -1,9 +1,12 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch.nn as nn from . import BasicModule -from ..module import Embedding -from ..module import Transformer as TransformerBlock +from module import Embedding +from module import Transformer as TransformerBlock -from ..utils import seq_len_to_mask +from utils import seq_len_to_mask class Transformer(BasicModule): diff --git a/src/deepke/relation_extraction/standard/models/BiLSTM.py b/src/deepke/relation_extraction/standard/models/BiLSTM.py index d3584fe..c9fc7c3 100644 --- a/src/deepke/relation_extraction/standard/models/BiLSTM.py +++ b/src/deepke/relation_extraction/standard/models/BiLSTM.py @@ -1,6 +1,9 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch.nn as nn from . import BasicModule -from ..module import Embedding, RNN +from module import Embedding, RNN class BiLSTM(BasicModule): diff --git a/src/deepke/relation_extraction/standard/models/Capsule.py b/src/deepke/relation_extraction/standard/models/Capsule.py index 889163f..ba62673 100644 --- a/src/deepke/relation_extraction/standard/models/Capsule.py +++ b/src/deepke/relation_extraction/standard/models/Capsule.py @@ -1,9 +1,12 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch from . import BasicModule -from ..module import Embedding, CNN -from ..module import Capsule as CapsuleLayer +from module import Embedding, CNN +from module import Capsule as CapsuleLayer -from ..utils import seq_len_to_mask, to_one_hot +from utils import seq_len_to_mask, to_one_hot class Capsule(BasicModule): diff --git a/src/deepke/relation_extraction/standard/models/GCN.py b/src/deepke/relation_extraction/standard/models/GCN.py index bf640d5..83d6909 100644 --- a/src/deepke/relation_extraction/standard/models/GCN.py +++ b/src/deepke/relation_extraction/standard/models/GCN.py @@ -1,9 +1,12 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch import torch.nn as nn from . import BasicModule -from ..module import Embedding -from ..module import GCN as GCNBlock -from ..utils import seq_len_to_mask +from module import Embedding +from module import GCN as GCNBlock +from utils import seq_len_to_mask class GCN(BasicModule): diff --git a/src/deepke/relation_extraction/standard/models/LM.py b/src/deepke/relation_extraction/standard/models/LM.py index 91db2ce..6b50f61 100644 --- a/src/deepke/relation_extraction/standard/models/LM.py +++ b/src/deepke/relation_extraction/standard/models/LM.py @@ -1,8 +1,11 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) from torch import nn from . import BasicModule -from ..module import RNN +from module import RNN from transformers import BertModel -from ..utils import seq_len_to_mask +from utils import seq_len_to_mask class LM(BasicModule): diff --git a/src/deepke/relation_extraction/standard/models/PCNN.py b/src/deepke/relation_extraction/standard/models/PCNN.py index 8d73603..7a45688 100644 --- a/src/deepke/relation_extraction/standard/models/PCNN.py +++ b/src/deepke/relation_extraction/standard/models/PCNN.py @@ -1,9 +1,12 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch import torch.nn as nn import torch.nn.functional as F from . import BasicModule -from ..module import Embedding, CNN -from ..utils import seq_len_to_mask +from module import Embedding, CNN +from utils import seq_len_to_mask class PCNN(BasicModule): diff --git a/src/deepke/relation_extraction/standard/models/Transformer.py b/src/deepke/relation_extraction/standard/models/Transformer.py index 1c57247..64af58b 100644 --- a/src/deepke/relation_extraction/standard/models/Transformer.py +++ b/src/deepke/relation_extraction/standard/models/Transformer.py @@ -1,8 +1,11 @@ +import os +import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))) import torch.nn as nn from . import BasicModule -from ..module import Embedding -from ..module import Transformer as TransformerBlock -from ..utils import seq_len_to_mask +from module import Embedding +from module import Transformer as TransformerBlock +from utils import seq_len_to_mask class Transformer(BasicModule):