20 lines
850 B
Python
20 lines
850 B
Python
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"""
|