首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
  • 您找到你想要的搜索结果了吗?
    是的
    没有找到

    TensorFlow2学习:RNN生成古诗词

    来源 | CSDN博客作者 | 蒋含竹责编 | 徐威龙 利用循环神经网络RNN可以做各种连续性数据的预测,其中生成古诗词是一件非常有趣的事,特此分享我的学习经验。...另外,我们抽取的原始数据还需要进行转码,才能喂给模型进行训练,该部分也封装在PoetryDataSet中 其代码如下 class PoetryDataSet: """ 古诗数据集生成器...generate_random_poem(tokenizer, model, text=""): """ 随机生成一首诗 :param tokenizer: 分词器 :param model: 古诗模型...:param text: 古诗的起始字符串,默认为空 :return: 一首古诗的字符串 """ # 将初始字符串转成token_ids,并去掉结束标记[END]...:param heads: 藏头诗的头 :return: 一首古诗的字符串 """ # token_ids,只包含[START]编号 token_ids = [

    1.6K30
    领券