Tensorflow + Django 打造诗词生成系统

最新推荐文章于 2024-05-20 10:51:01 发布

碳纤维石头君

最新推荐文章于 2024-05-20 10:51:01 发布

阅读量851

点赞数 2

分类专栏： Tensorflow实战系列教程文章标签： tensorflow 神经网络自然语言处理

本文链接：https://blog.csdn.net/h12321hzhy/article/details/104405084

版权

Tensorflow实战系列教程专栏收录该内容

5 篇文章 0 订阅

订阅专栏

用Tensorflow + Django打造一款诗词生成系统

安装Tensorflow,安装Tensorflow

pip install tensorflow==1.15.0
pip install Django

创建Django默认框架

django-admin startproject aitemplate

前端Html页面

<!DOCTYPE html>
<html>
<head> 
<meta charset="utf-8"> 
<!-- 欢迎关注B站 碳纤维石头君  -->
<title>欢迎关注B站 碳纤维石头君</title> 
</head>
<body>
<script type="text/javascript" color="0,0,255" opacity='0.7' zIndex="-2" count="99" src="//cdn.bootcss.com/canvas-nest.js/1.0.1/canvas-nest.min.js"></script>
<form action="update_poem" method="get">
    {% csrf_token %}
    请输入第一句诗词: <input type="text" name="FirstName" value="{{ inputs }}"><br>
    <p></p>
    <input type="submit" value="提交">
    <style> 
    body{ text-align:center} 
    .div{ margin:0 auto; width:400px; height:100px; border:1px solid #F00} 
    </style> 
</form>
<h1>{{ result }}</h1>

</body>
</html>

LSTM诗词生成诗词 + 藏头诗等

import tensorflow as tf
from rnn.read_utils import TextConverter
from rnn.model import CharRNN
import os
#from IPython import embed



class PoemAdapter:
    def __init__(self):
        self.checkpoint_path = "rnn/model/default"
        self.converter = TextConverter(filename="rnn/model/default/converter.pkl")
        self.checkpoint_path =tf.train.latest_checkpoint(self.checkpoint_path)
        self.model = CharRNN(self.converter.vocab_size, lstm_size=128, num_layers=2,use_embedding=True,embedding_size=128)
        self.model.load(self.checkpoint_path)

    def gen_poem(self,starts):
        start = self.converter.text_to_arr(starts)
        arr = self.model.build_cangtoushi(start, self.converter.vocab_size)
        return self.converter.arr_to_text(arr)

    def gen_poem1(self,starts):
        start = self.converter.text_to_arr(starts)
        arr = self.model.build_sample(start, self.converter.vocab_size)
        return self.converter.arr_to_text(arr)

global poemfactory
poemfactory = PoemAdapter()

Django 进行算法绑定

from django.shortcuts import render
from django.http import HttpResponse
from django.http import JsonResponse

from rnn import sample
def interface(request):
    context = {}
    return render(request,"index.html",context)


def get_poem(request):
    request.encoding='utf-8'
    input_FirstName = request.GET['FirstName']
    context = {}
    context['inputs'] = input_FirstName
    #设置为藏头诗模式
    context['result'] = sample.poemfactory.gen_poem(input_FirstName)
    #设置为生成诗词模式
    #context['result'] = sample.poemfactory.gen_poem1(input_FirstName)
    return render(request,"index.html",context)

OK，一套完整的在web诗词系统，就做好啦，效果图，顺带一块为武汉加油：

完整的视频课程，请关注我的个人视频号，我是石头君:

https://www.bilibili.com/video/av90216321/

https://www.bilibili.com/video/av90216321/

碳纤维石头君

关注

2
点赞
踩
4

收藏

觉得还不错? 一键收藏
打赏
2
评论
Tensorflow + Django 打造诗词生成系统

用Tensorflow + Django打造一款诗词生成系统安装Tensorflow,安装Tensorflowpip install tensorflow==1.15.0pip install Django创建Django默认框架django-admin startproject aitemplate前端Html页面<!DOCTYPE html>&...
复制链接

扫一扫