config.py

import os
import pathlib
this_path = pathlib.Path(__file__).parent.absolute()

class configurations(object):
    def __init__(self):
        self.batch_size = 64
        self.emb_dim = 256
        self.hid_dim = 512
        self.n_layers = 3
        self.dropout = 0.5
        self.learning_rate = 0.00005
        self.max_output_len = 100                # 最後輸出句子的最大長度
        self.num_steps = 50000                # 總訓練次數
        self.store_steps = 1000                # 訓練多少次後須儲存模型
        self.summary_steps = 1000              # 訓練多少次後須檢驗是否有overfitting
        self.load_model = True               # 是否需載入模型
        self.store_model_path = os.path.join(this_path, "ckpt")      # 儲存模型的位置
        # 載入模型的位置 e.g. "./ckpt/model_{step}"
        self.load_model_path = os.path.join(this_path, "ckpt/model_10000.ckpt")
        self.data_path = os.path.join(this_path, "data")          # 資料存放的位置
        self.attention = True                # 是否使用 Attention Mechanism