15 lines
563 B
Python
15 lines
563 B
Python
from pathlib import Path
|
|
|
|
DIRTY_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'poetry.txt'
|
|
"""脏的（未清洗的）古诗数据的路径"""
|
|
CLEAN_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'poetry.pickle'
|
|
"""干净的（已经清洗过的）古诗数据的路径"""
|
|
|
|
SAVED_MODULE_PATH: Path = Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'
|
|
"""训练完毕的模型进行保存的路径"""
|
|
|
|
N_EPOCH: int = 10
|
|
"""训练时的epoch"""
|
|
N_BATCH_SIZE: int = 16
|
|
"""训练时的batch size"""
|