finish exp2

exp3/modified/settings.py

@@ -1,12 +1,14 @@
 from pathlib import Path
 
-BATCH_SIZE: int = 16
-"""训练的batch size"""
+DIRTY_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'poetry.txt'
+"""脏的（未清洗的）古诗数据的路径"""
+CLEAN_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'poetry.pickle'
+"""干净的（已经清洗过的）古诗数据的路径"""
 
-def get_saved_model_path() -> Path:
-    """
-    获取训练完毕的模型进行保存的路径。
-    
-    :return: 模型参数保存的路径。
-    """
-    return Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'
+SAVED_MODULE_PATH: Path = Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'
+"""训练完毕的模型进行保存的路径"""
+
+N_EPOCH: int = 10
+"""训练时的epoch"""
+N_BATCH_SIZE: int = 16
+"""训练时的batch size"""