# -*- coding: utf-8 -*-
# @File    : settings.py
# @Author  : AaronJny
# @Time    : 2019/12/25
# @Desc    :

# 预训练的模型参数
CONFIG_PATH = '/home/bitterteaer/disk_extends/PycharmProjects/古诗生成器/chinese_L-12_H-768_A-12/bert_config.json'
CHECKPOINT_PATH = '/home/bitterteaer/disk_extends/PycharmProjects/古诗生成器/chinese_L-12_H-768_A-12/bert_model.ckpt'
DICT_PATH = '/home/bitterteaer/disk_extends/PycharmProjects/古诗生成器/chinese_L-12_H-768_A-12/vocab.txt'
# 禁用词，包含如下字符的唐诗将被忽略
DISALLOWED_WORDS = ['（', '）', '(', ')', '__', '《', '》', '【', '】', '[', ']']
# 句子最大长度
MAX_LEN = 64
# 最小词频
MIN_WORD_FREQUENCY = 8
# 训练的batch size
BATCH_SIZE = 32
# 数据集路径
DATASET_PATH = './poetry.txt'
# 每个epoch训练完成后，随机生成SHOW_NUM首古诗作为展示
SHOW_NUM = 5
# 共训练多少个epoch
TRAIN_EPOCHS = 20
# 最佳权重保存路径
BEST_MODEL_PATH = './best_model.weights'