from pathlib import Path

POETRY_MAX_LEN: int = 64
"""古诗词句子最大允许长度（该长度包含首尾填充的特殊字符），超过该长度的诗句将被删除。"""
POETRY_MIN_WORD_FREQ: int = 8
"""古诗词最小允许词频，小于该词频的词将在编解码时被视为[UNK]生僻字。"""

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_MODEL_PATH: Path = Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'
"""训练完毕的模型进行保存的路径"""

N_EPOCH: int = 10
"""训练时的epoch"""
N_BATCH_SIZE: int = 50
"""训练时的batch size"""