来看看程序

from gensim.models import Word2Vec  # 词向量
from random import choice
from os.path import exists
from warnings import filterwarnings
filterwarnings('ignore')  # 不打印警告class CONF:path = '古诗词.txt'  # 语料路径window = 16  # 滑窗大小min_count = 60  # 过滤低频字size = 125  # 词向量维度topn = 14  # 生成诗词的开放度model_path = 'word2vec'  # 模型路径class Model:def __init__(self, window, topn, model):self.window = windowself.topn = topnself.model = model  # 词向量模型self.chr_dict = model.wv.index2word  # 字典"""模型初始化"""@classmethoddef initialize(cls, config):if exists(config.model_path):# 模型读取model = Word2Vec.load(config.model_path)else:# 语料读取with open(config.path, encoding='utf-8') as f:ls_of_ls_of_c = [list(line.strip()) for line in f]# 模型训练和保存model = Word2Vec(ls_of_ls_of_c, config.size,window=config.window, min_count=config.min_count)model.save(config.model_path)return cls(config.window, config.topn, model)"""古诗词生成"""def poem_generator(self, title, form):# list_of_tuples --> list_of_strings 并过滤标点符号filter = lambda lst: [t[0] for t in lst if t[0] not in ['，', '。']]# 标题补全if len(title) < 4:if not title:title += choice(self.chr_dict)for _ in range(4 - len(title)):chrs = self.model.similar_by_word(title[-1], self.topn // 2)chrs = filter(chrs)char = choice([c for c in chrs if c not in title])title += char# 文本生成seq = list(title)for i in range(form[0]):for _ in range(form[1]):chrs = self.model.predict_output_word(seq[-self.window:], max(self.topn, len(seq) + 1))chrs = filter(chrs)char = choice([c for c in chrs if c not in seq[len(title):]])seq.append(char)seq.append('，' if i % 2 == 0 else '。')# 返回标题+主体length = form[0] * (form[1] + 1)title = '《%s》' % ''.join(seq[:-length])poem = ''.join(seq[-length:])return title + '\n' + poemdef main(config=CONF):form = {'五言绝句': (4, 5), '七言绝句': (4, 7), '对联': (2, 9)}m = Model.initialize(config)while True:title = input('输入标题：').strip()try:poem = m.poem_generator(title, form['五言绝句'])print('\033[031m%s\033[0m' % poem)  # redpoem = m.poem_generator(title, form['七言绝句'])print('\033[033m%s\033[0m' % poem)  # yellowpoem = m.poem_generator(title, form['对联'])print('\033[036m%s\033[0m' % poem)  # purpleprint()except:passif __name__ == '__main__':main()