add cote-dp & seabsa16_phns dataset docs (PaddlePaddle#436)

* add cote-dp & seabsa16_phns dataset docs
wangxicoding · May 24, 2021 · c3d1df6 · c3d1df6
1 parent 9feb941
commit c3d1df6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docs/data_prepare/dataset_list.rst b/docs/data_prepare/dataset_list.rst
@@ -48,6 +48,11 @@ PaddleNLP提供了以下数据集的快速读取API，实际使用时请根据
 +---------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------------+-------------------------------------------------------+
 | `ChnSentiCorp <https://github.com/SophonPlus/ChineseNlpCorpus/blob/master/datasets/ChnSentiCorp_htl_all/intro.ipynb>`__   | 中文评论情感分析语料                                                                                                                     | ``paddlenlp.datasets.load_dataset('chnsenticorp')``   |
 +---------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------------+-------------------------------------------------------+
+| `COTE-DP<https://aistudio.baidu.com/aistudio/competition/detail/50/?isFromLuge=1>`__   | 中文观点抽取语料                                                                                                                     | ``paddlenlp.datasets.load_dataset('cote', 'dp')``   |
++---------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------------+-------------------------------------------------------+
+| `SE-ABSA16_PHNS<https://aistudio.baidu.com/aistudio/competition/detail/50/?isFromLuge=1>`__   | 中文评价对象级情感分析语料                                                                                                                     | ``paddlenlp.datasets.load_dataset('seabsa16', 'phns')``   |
++---------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------------+-------------------------------------------------------+
+
 
 序列标注
 --------

diff --git a/examples/language_model/bert/run_pretrain.py b/examples/language_model/bert/run_pretrain.py
@@ -333,8 +333,7 @@ def do_train(args):
     for epoch in range(args.num_train_epochs):
         files = [
             os.path.join(args.input_dir, f) for f in os.listdir(args.input_dir)
-            if os.path.isfile(os.path.join(args.input_dir, f)) and "training" in
-            f
+            if os.path.isfile(os.path.join(args.input_dir, f)) and "train" in f
         ]
         files.sort()
         num_files = len(files)