andre/chinese-poetry
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update data_loader.py

Browse Source
This commit is contained in:
SanBingYouYong-S.Zhang
2023-05-10 17:34:45 +01:00
committed by GitHub
parent aa65d2fa5e
commit 717029ade7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
loader/data_loader.py
View File

@ -5,7 +5,7 @@ import os
DATAS_CONFIG = "./loader/datas.json"
class INGDataLoader():
class PlainDataLoader():
def __init__(self, config_path: str=DATAS_CONFIG) -> None:
self._path = config_path
with open(config_path, 'r', encoding='utf-8') as config:
@ -58,7 +58,7 @@ class INGDataLoader():
if __name__ == "__main__":
loader = INGDataLoader()
loader = PlainDataLoader()
print(loader.id_table)
print(
loader.body_extractor("wudai-huajianji")[-1]