20 lines
536 B
Python
20 lines
536 B
Python
|
#ANLI College of Artificial Intelligence
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 禁用词，包含如下字符的唐诗将被忽略
|
|||
|
|
DISALLOWED_WORDS = ['（', '）', '(', ')', '__', '《', '》', '【', '】', '[', ']']
|
|||
|
|
# 句子最大长度
|
|||
|
|
MAX_LEN = 64
|
|||
|
|
# 最小词频
|
|||
|
|
MIN_WORD_FREQUENCY = 8
|
|||
|
|
# 训练的batch size
|
|||
|
|
BATCH_SIZE = 16
|
|||
|
|
# 数据集路径
|
|||
|
|
DATASET_PATH = './poetry.txt'
|
|||
|
|
# 每个epoch训练完成后，随机生成SHOW_NUM首古诗作为展示
|
|||
|
|
SHOW_NUM = 5
|
|||
|
|
# 共训练多少个epoch
|
|||
|
|
TRAIN_EPOCHS = 10
|
|||
|
|
# 最佳权重保存路径
|
|||
|
|
BEST_MODEL_PATH = './best_model.h5'
|