setup of pypi

yongzhuo · Sep 27, 2021 · acb5cdb · acb5cdb
1 parent 0240a44
commit acb5cdb
Show file tree

Hide file tree

Showing 60 changed files with 76 additions and 40 deletions.
diff --git a/README.md b/README.md
@@ -109,9 +109,11 @@ import json
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # 分类下的引入, pytorch_textclassification
+from tcConfig import model_config
+os.environ["CUDA_VISIBLE_DEVICES"] = model_config.get("CUDA_VISIBLE_DEVICES", "0")
 from tcTools import get_current_time
 from tcRun import TextClassification
 from tcConfig import model_config
@@ -157,11 +159,13 @@ import json
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)
 # 分类下的引入, pytorch_textclassification
+from tcConfig import model_config
+os.environ["CUDA_VISIBLE_DEVICES"] = model_config.get("CUDA_VISIBLE_DEVICES", "0")
 from slTools import get_current_time
 from slRun import SequenceLabeling
 from slConfig import model_config

diff --git a/output/__init__.py → pytorch_nlu/__init__.py b/output/__init__.py → pytorch_nlu/__init__.py
@@ -1,5 +1,5 @@
 # !/usr/bin/python
 # -*- coding: utf-8 -*-
-# @time    : 2021/7/27 19:28
+# @time    : 2021/9/27 23:18
 # @author  : Mo
 # @function:
diff --git a/corpus/__init__.py → pytorch_nlu/corpus/__init__.py b/corpus/__init__.py → pytorch_nlu/corpus/__init__.py
diff --git a/...r_china_people_daily_1998_conll/README.md → ...r_china_people_daily_1998_conll/README.md b/...r_china_people_daily_1998_conll/README.md → ...r_china_people_daily_1998_conll/README.md
diff --git a/...r_china_people_daily_1998_conll/dev.conll → ...r_china_people_daily_1998_conll/dev.conll b/...r_china_people_daily_1998_conll/dev.conll → ...r_china_people_daily_1998_conll/dev.conll
diff --git a/..._china_people_daily_1998_conll/test.conll → ..._china_people_daily_1998_conll/test.conll b/..._china_people_daily_1998_conll/test.conll → ..._china_people_daily_1998_conll/test.conll
diff --git a/...china_people_daily_1998_conll/train.conll → ...china_people_daily_1998_conll/train.conll b/...china_people_daily_1998_conll/train.conll → ...china_people_daily_1998_conll/train.conll
diff --git a/...er_china_people_daily_1998_span/README.md → ...er_china_people_daily_1998_span/README.md b/...er_china_people_daily_1998_span/README.md → ...er_china_people_daily_1998_span/README.md
diff --git a/...ner_china_people_daily_1998_span/dev.span → ...ner_china_people_daily_1998_span/dev.span b/...ner_china_people_daily_1998_span/dev.span → ...ner_china_people_daily_1998_span/dev.span
diff --git a/...er_china_people_daily_1998_span/test.span → ...er_china_people_daily_1998_span/test.span b/...er_china_people_daily_1998_span/test.span → ...er_china_people_daily_1998_span/test.span
diff --git a/...r_china_people_daily_1998_span/train.span → ...r_china_people_daily_1998_span/train.span b/...r_china_people_daily_1998_span/train.span → ...r_china_people_daily_1998_span/train.span
diff --git a/corpus/text_classification/school/README.md → ...rpus/text_classification/school/README.md b/corpus/text_classification/school/README.md → ...rpus/text_classification/school/README.md
diff --git a/corpus/text_classification/school/dev.json → ...orpus/text_classification/school/dev.json b/corpus/text_classification/school/dev.json → ...orpus/text_classification/school/dev.json
diff --git a/corpus/text_classification/school/test.json → ...rpus/text_classification/school/test.json b/corpus/text_classification/school/test.json → ...rpus/text_classification/school/test.json
diff --git a/corpus/text_classification/school/train.json → ...pus/text_classification/school/train.json b/corpus/text_classification/school/train.json → ...pus/text_classification/school/train.json
diff --git a/corpus/text_classification/tnews/README.md → ...orpus/text_classification/tnews/README.md b/corpus/text_classification/tnews/README.md → ...orpus/text_classification/tnews/README.md
diff --git a/corpus/text_classification/tnews/dev.json → ...corpus/text_classification/tnews/dev.json b/corpus/text_classification/tnews/dev.json → ...corpus/text_classification/tnews/dev.json
diff --git a/corpus/text_classification/tnews/test.json → ...orpus/text_classification/tnews/test.json b/corpus/text_classification/tnews/test.json → ...orpus/text_classification/tnews/test.json
diff --git a/corpus/text_classification/tnews/train.json → ...rpus/text_classification/tnews/train.json b/corpus/text_classification/tnews/train.json → ...rpus/text_classification/tnews/train.json
diff --git a/pytorch_nlu/output/__init__.py b/pytorch_nlu/output/__init__.py
@@ -0,0 +1,5 @@
+# !/usr/bin/python
+# -*- coding: utf-8 -*-
+# @time    : 2021/9/27 23:32
+# @author  : Mo
+# @function:
diff --git a/pytorch_sequencelabeling/README.md → ...ch_nlu/pytorch_sequencelabeling/README.md b/pytorch_sequencelabeling/README.md → ...ch_nlu/pytorch_sequencelabeling/README.md
@@ -85,7 +85,7 @@ import json
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/pytorch_sequencelabeling/__init__.py → ..._nlu/pytorch_sequencelabeling/__init__.py b/pytorch_sequencelabeling/__init__.py → ..._nlu/pytorch_sequencelabeling/__init__.py
diff --git a/pytorch_sequencelabeling/slAdversarial.py → ...pytorch_sequencelabeling/slAdversarial.py b/pytorch_sequencelabeling/slAdversarial.py → ...pytorch_sequencelabeling/slAdversarial.py
diff --git a/pytorch_sequencelabeling/slConfig.py → ..._nlu/pytorch_sequencelabeling/slConfig.py b/pytorch_sequencelabeling/slConfig.py → ..._nlu/pytorch_sequencelabeling/slConfig.py
@@ -5,6 +5,8 @@
 # @function: config of sequence-labeling, 超参数/类
 
 
+import os
+os.environ["USE_TORCH"] = "1"
 from transformers import BertTokenizer, RobertaTokenizer, AlbertTokenizer, XLNetTokenizer, ElectraTokenizer, XLMTokenizer, AutoTokenizer
 from transformers import BertConfig, RobertaConfig, AlbertConfig, XLNetConfig, ElectraConfig, XLMConfig, AutoConfig
 from transformers import BertModel, RobertaModel, AlbertModel, XLNetModel, ElectraModel, XLMModel, AutoModel

diff --git a/pytorch_sequencelabeling/slData.py → ...ch_nlu/pytorch_sequencelabeling/slData.py b/pytorch_sequencelabeling/slData.py → ...ch_nlu/pytorch_sequencelabeling/slData.py
diff --git a/pytorch_sequencelabeling/slGraph.py → ...h_nlu/pytorch_sequencelabeling/slGraph.py b/pytorch_sequencelabeling/slGraph.py → ...h_nlu/pytorch_sequencelabeling/slGraph.py
diff --git a/pytorch_sequencelabeling/slLayer.py → ...h_nlu/pytorch_sequencelabeling/slLayer.py b/pytorch_sequencelabeling/slLayer.py → ...h_nlu/pytorch_sequencelabeling/slLayer.py
diff --git a/pytorch_sequencelabeling/slOffice.py → ..._nlu/pytorch_sequencelabeling/slOffice.py b/pytorch_sequencelabeling/slOffice.py → ..._nlu/pytorch_sequencelabeling/slOffice.py
diff --git a/pytorch_sequencelabeling/slPredict.py → ...nlu/pytorch_sequencelabeling/slPredict.py b/pytorch_sequencelabeling/slPredict.py → ...nlu/pytorch_sequencelabeling/slPredict.py
@@ -10,7 +10,8 @@
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "."))
 sys.path.append(path_root)
-# os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+from tcConfig import model_config
+os.environ["CUDA_VISIBLE_DEVICES"] = model_config.get("CUDA_VISIBLE_DEVICES", "0")
 from slConfig import _SL_MODEL_SOFTMAX, _SL_MODEL_GRID, _SL_MODEL_SPAN, _SL_MODEL_CRF
 from slConfig import _SL_DATA_CONLL, _SL_DATA_SPAN
 from slTools import get_logger, load_json

diff --git a/pytorch_sequencelabeling/slRun.py → ...rch_nlu/pytorch_sequencelabeling/slRun.py b/pytorch_sequencelabeling/slRun.py → ...rch_nlu/pytorch_sequencelabeling/slRun.py
@@ -12,6 +12,8 @@
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "."))
 sys.path.append(path_root)
 
+from tcConfig import model_config
+os.environ["CUDA_VISIBLE_DEVICES"] = model_config.get("CUDA_VISIBLE_DEVICES", "0")
 from slConfig import _SL_MODEL_SOFTMAX, _SL_MODEL_GRID, _SL_MODEL_SPAN, _SL_MODEL_CRF
 from slConfig import _SL_DATA_CONLL, _SL_DATA_SPAN
 from slTools import get_logger

diff --git a/pytorch_sequencelabeling/slTools.py → ...h_nlu/pytorch_sequencelabeling/slTools.py b/pytorch_sequencelabeling/slTools.py → ...h_nlu/pytorch_sequencelabeling/slTools.py
diff --git a/pytorch_sequencelabeling/slTqdm.py → ...ch_nlu/pytorch_sequencelabeling/slTqdm.py b/pytorch_sequencelabeling/slTqdm.py → ...ch_nlu/pytorch_sequencelabeling/slTqdm.py
diff --git a/pytorch_textclassification/README.md → ..._nlu/pytorch_textclassification/README.md b/pytorch_textclassification/README.md → ..._nlu/pytorch_textclassification/README.md
@@ -68,7 +68,7 @@ import json
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # 分类下的引入, pytorch_textclassification
 from tcTools import get_current_time

diff --git a/pytorch_textclassification/__init__.py → ...lu/pytorch_textclassification/__init__.py b/pytorch_textclassification/__init__.py → ...lu/pytorch_textclassification/__init__.py
diff --git a/pytorch_textclassification/tcAdversarial.py → ...torch_textclassification/tcAdversarial.py b/pytorch_textclassification/tcAdversarial.py → ...torch_textclassification/tcAdversarial.py
diff --git a/pytorch_textclassification/tcConfig.py → ...lu/pytorch_textclassification/tcConfig.py b/pytorch_textclassification/tcConfig.py → ...lu/pytorch_textclassification/tcConfig.py
@@ -4,7 +4,8 @@
 # @author  : Mo
 # @function: config of transformers and graph-model
 
-
+import os
+os.environ["USE_TORCH"] = "1"
 from transformers import BertTokenizer, RobertaTokenizer, AlbertTokenizer, XLNetTokenizer, ElectraTokenizer, XLMTokenizer, AutoTokenizer
 from transformers import BertConfig, RobertaConfig, AlbertConfig, XLNetConfig, ElectraConfig, XLMConfig, AutoConfig
 from transformers import BertModel, RobertaModel, AlbertModel, XLNetModel, ElectraModel, XLMModel, AutoModel
@@ -36,7 +37,7 @@
 
 # model算法超参数
 model_config = {
-    "CUDA_VISIBLE_DEVICES": "1",  # 环境, GPU-CPU, "-1"/"0"/"1"/"2"...
+    "CUDA_VISIBLE_DEVICES": "0",  # 环境, GPU-CPU, "-1"/"0"/"1"/"2"...
     "output_hidden_states": None,  # [6,11]  # 输出层, 即取第几层transformer的隐藏输出, list
     "pretrained_model_name_or_path": "",  # 预训练模型地址
     "model_save_path": "save_path",  # 训练模型保存-训练完毕模型目录

diff --git a/pytorch_textclassification/tcData.py → ..._nlu/pytorch_textclassification/tcData.py b/pytorch_textclassification/tcData.py → ..._nlu/pytorch_textclassification/tcData.py
diff --git a/pytorch_textclassification/tcGraph.py → ...nlu/pytorch_textclassification/tcGraph.py b/pytorch_textclassification/tcGraph.py → ...nlu/pytorch_textclassification/tcGraph.py
diff --git a/pytorch_textclassification/tcLayer.py → ...nlu/pytorch_textclassification/tcLayer.py b/pytorch_textclassification/tcLayer.py → ...nlu/pytorch_textclassification/tcLayer.py
@@ -188,14 +188,14 @@ def __init__(self, reduction="mean", inf=1e12):
 
     def forward(self, logits, labels):
         logits = (1 - 2 * labels) * logits              # <3, 4>
-        logits_neg = logits - labels * self.inf         # <3, 4>
-        logits_pos = logits - (1 - labels) * self.inf   # <3, 4>
+        logits_neg = logits - labels * self.inf         # <3, 4>, 减去选中多标签的index
+        logits_pos = logits - (1 - labels) * self.inf   # <3, 4>, 减去其他不需要的多标签Index
         zeros = torch.zeros_like(logits[..., :1])       # <3, 1>
         logits_neg = torch.cat([logits_neg, zeros], dim=-1)  # <3, 5>
         logits_pos = torch.cat([logits_pos, zeros], dim=-1)  # <3, 5>
         neg_loss = torch.logsumexp(logits_neg, dim=-1)       # <3, >
         pos_loss = torch.logsumexp(logits_pos, dim=-1)       # <3, >
-        loss = neg_loss + pos_loss
+        loss = neg_loss + pos_loss                           # pos比零大, neg比零小
         if "mean" == self.reduction:
             loss = loss.mean()
         else:

diff --git a/pytorch_textclassification/tcOffice.py → ...lu/pytorch_textclassification/tcOffice.py b/pytorch_textclassification/tcOffice.py → ...lu/pytorch_textclassification/tcOffice.py
diff --git a/pytorch_textclassification/tcPredict.py → ...u/pytorch_textclassification/tcPredict.py b/pytorch_textclassification/tcPredict.py → ...u/pytorch_textclassification/tcPredict.py
@@ -10,7 +10,8 @@
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "."))
 sys.path.append(path_root)
-# os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+from tcConfig import model_config
+os.environ["CUDA_VISIBLE_DEVICES"] = model_config.get("CUDA_VISIBLE_DEVICES", "0")
 from tcTools import get_logger, load_json
 from tcOffice import Office
 from tcData import Corpus
@@ -60,7 +61,8 @@ def predict(self, texts, logits_type="sigmoid"):
 
 
 if __name__ == "__main__":
-
+    # BERT-base = 8109M
+    # path_config = "../output/text_classification/model_BERT/tc.config"
     path_config = "../output/text_classification/model_ERNIE/tc.config"
     tcp = TextClassificationPredict(path_config)
     texts = [{"text": "平乐县，古称昭州，隶属于广西壮族自治区桂林市，位于广西东北部，桂林市东南部，东临钟山县，南接昭平，西北毗邻阳朔，北连恭城，总面积1919.34平方公里。"},

diff --git a/pytorch_textclassification/tcRun.py → ...h_nlu/pytorch_textclassification/tcRun.py b/pytorch_textclassification/tcRun.py → ...h_nlu/pytorch_textclassification/tcRun.py
@@ -10,6 +10,8 @@
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "."))
 sys.path.append(path_root)
+from tcConfig import model_config
+os.environ["CUDA_VISIBLE_DEVICES"] = model_config.get("CUDA_VISIBLE_DEVICES", "0")
 from tcConfig import _TC_MULTI_CLASS, _TC_MULTI_LABEL
 from tcTools import get_logger
 from tcOffice import Office
@@ -123,11 +125,10 @@ def eval(self):
         save_steps = 320  # 存储步数
         ee = 0
     # 训练-验证语料地址, 可以只输入训练地址
-    # path_corpus = path_root + "/corpus/text_classification/school"
-    path_corpus = path_root + "/corpus/text_classification/org_tnews"
-
-    path_train = os.path.join(path_corpus, "train.json")
-    path_dev = os.path.join(path_corpus, "dev.json")
+    path_corpus = path_root + "/corpus/text_classification/org_multi-label_school"
+    # path_corpus = path_root + "/corpus/text_classification/org_tnews"
+    path_train = os.path.join(path_corpus, "train.json.augment")
+    path_dev = os.path.join(path_corpus, "dev.json.augment")
     model_config["evaluate_steps"] = evaluate_steps  # 评估步数
     model_config["save_steps"] = save_steps  # 存储步数
     model_config["path_train"] = path_train
@@ -145,17 +146,17 @@ def eval(self):
         "ROBERTA": pretrained_model_dir + "/chinese_roberta_wwm_ext_pytorch",
         "ALBERT": pretrained_model_dir + "/albert_base_v1",
         "XLNET": pretrained_model_dir + "/chinese_xlnet_mid_pytorch",
-        "ERNIE": pretrained_model_dir + "/ERNIE_stable-1.0.1-pytorch",
-        # "ERNIE": pretrained_model_dir + "/ernie-tiny",
+        # "ERNIE": pretrained_model_dir + "/ERNIE_stable-1.0.1-pytorch",
+        "ERNIE": pretrained_model_dir + "/ernie-tiny",
         "BERT": pretrained_model_dir + "/bert-base-chinese",
     }
-    idx = 0  # 选择的预训练模型类型---model_type
+    idx = 1  # 选择的预训练模型类型---model_type
     model_config["pretrained_model_name_or_path"] = pretrained_model_name_or_path[model_type[idx]]
     # model_config["model_save_path"] = "../output/text_classification/model_{}".format(model_type[idx] + "_" + str(get_current_time()))
     model_config["model_save_path"] = "../output/text_classification/model_{}".format(model_type[idx])
     model_config["model_type"] = model_type[idx]
 
-    os.environ["CUDA_VISIBLE_DEVICES"] = str(model_config["CUDA_VISIBLE_DEVICES"])
+    # os.environ["CUDA_VISIBLE_DEVICES"] = str(model_config["CUDA_VISIBLE_DEVICES"])
 
     # main
     lc = TextClassification(model_config)

diff --git a/pytorch_textclassification/tcTools.py → ...nlu/pytorch_textclassification/tcTools.py b/pytorch_textclassification/tcTools.py → ...nlu/pytorch_textclassification/tcTools.py
diff --git a/pytorch_textclassification/tcTqdm.py → ..._nlu/pytorch_textclassification/tcTqdm.py b/pytorch_textclassification/tcTqdm.py → ..._nlu/pytorch_textclassification/tcTqdm.py
diff --git a/pytorch_nlu/version.py b/pytorch_nlu/version.py
@@ -0,0 +1,8 @@
+# !/usr/bin/python
+# -*- coding: utf-8 -*-
+# @time    : 2020/12/21 22:24
+# @author  : Mo
+# @function: version of Pytorch-NLU
+
+
+__version__ = "0.0.1"
diff --git a/setup.py b/setup.py
@@ -5,6 +5,7 @@
 # @function :setup of Pytorch-NLU
 
 
+from pytorch_nlu.version import __version__
 from setuptools import find_packages, setup
 import codecs
 
@@ -23,23 +24,32 @@
     install_requires = list(map(lambda x: x.strip(), reader.readlines()))
 
 setup(name=NAME,
-        version="0.0.1",
+        version=__version__,
         description=DESCRIPTION,
         long_description=long_description,
         long_description_content_type="text/markdown",
         author=AUTHOR,
         author_email=EMAIL,
         url=URL,
-        packages=find_packages(exclude=("test")),
+        packages=find_packages(),
         install_requires=install_requires,
-        include_package_data=True,
+        package_data={"pytorch_nlu": ["*.*", "corpus/*",
+                                      "pytorch_textclassification/*"
+                                      "pytorch_sequencelabeling/*",
+                                      "corpus/text_classification/*",
+                                      "corpus/sequence_labeling/*",
+                                      "corpus/text_classification/school/*",
+                                      "corpus/text_classification/tnews/*",
+                                      "corpus/sequence_labeling/ner_china_people_daily_1998_conll/*",
+                                      "corpus/sequence_labeling/ner_china_people_daily_1998_span/*",]},
         license=LICENSE,
-        classifiers=["License :: OSI Approved :: MIT License",
+        classifiers=["License :: OSI Approved :: Apache License",
                      "Programming Language :: Python :: 3.4",
                      "Programming Language :: Python :: 3.5",
                      "Programming Language :: Python :: 3.6",
                      "Programming Language :: Python :: 3.7",
                      "Programming Language :: Python :: 3.8",
+                     "Programming Language :: Python :: 3.9",
                      "Programming Language :: Python :: Implementation :: CPython",
                      "Programming Language :: Python :: Implementation :: PyPy"],)
 

diff --git a/test/sl/tet_sl_base_crf.py b/test/sl/tet_sl_base_crf.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_crf_ernie.py b/test/sl/tet_sl_base_crf_ernie.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_data_conll.py b/test/sl/tet_sl_base_data_conll.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_data_span.py b/test/sl/tet_sl_base_data_span.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_grid.py b/test/sl/tet_sl_base_grid.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_predict.py b/test/sl/tet_sl_base_predict.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_softmax.py b/test/sl/tet_sl_base_softmax.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/sl/tet_sl_base_span.py b/test/sl/tet_sl_base_span.py
@@ -12,7 +12,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-path_sys = os.path.join(path_root, "pytorch_sequencelabeling")
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_sequencelabeling")
 sys.path.append(path_sys)
 print(path_root)
 print(path_sys)

diff --git a/test/tc/tet_tc_base_multi_class.py b/test/tc/tet_tc_base_multi_class.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # 分类下的引入, pytorch_textclassification
 from tcTools import get_current_time

diff --git a/test/tc/tet_tc_base_multi_label.py b/test/tc/tet_tc_base_multi_label.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # 分类下的引入, pytorch_textclassification
 from tcTools import get_current_time

diff --git a/test/tc/tet_tc_base_multi_label_focalloss.py b/test/tc/tet_tc_base_multi_label_focalloss.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # 分类下的引入, pytorch_textclassification
 from tcTools import get_current_time

diff --git a/test/tc/tet_tc_base_multi_label_isadv.py b/test/tc/tet_tc_base_multi_label_isadv.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # 分类下的引入, pytorch_textclassification
 from tcTools import get_current_time

diff --git a/test/tc/tet_tc_base_predict_multiclass.py b/test/tc/tet_tc_base_predict_multiclass.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 from tcPredict import TextClassificationPredict

diff --git a/test/tc/tet_tc_base_predict_multilabel.py b/test/tc/tet_tc_base_predict_multilabel.py
@@ -11,7 +11,7 @@
 import sys
 import os
 path_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
-sys.path.append(os.path.join(path_root, "pytorch_textclassification"))
+path_sys = os.path.join(path_root, "pytorch_nlu", "pytorch_textclassification")
 print(path_root)
 # os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 from tcPredict import TextClassificationPredict