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"""