【源码】基于Python和PyTorch的藏头诗生成器

项目简介

本项目是基于Python和PyTorch框架开发的藏头诗生成系统。借助深度学习模型，尤其是长短期记忆（LSTM）和Transformer模型，可依据用户输入的每句诗的第一个字（头韵）生成完整的藏头诗。项目包含训练模型、处理数据的代码，还配备图形用户界面（GUI）展示生成结果。

项目的主要特性和功能

1. 数据处理与模型训练：dataloader.py模块实现数据加载、分词、建立词汇表、划分训练/验证/测试集等功能。model.py模块定义LSTM和Transformer两种模型，用于训练数据并生成藏头诗。
2. GUI界面：MainWidget.py和MainWidget.ui文件定义GUI界面的结构与布局，通过testui.py文件启动，方便用户与模型交互。
3. 用户交互：用户既能通过命令行输入头韵，也能在GUI中输入，程序会生成相应藏头诗并显示。
4. 多模式支持：支持五言绝句、七言绝句、五言律诗、七言律诗等多种模式的藏头诗生成，用户可按需选择。

安装使用步骤

1. 环境准备：确保已安装Python和PyTorch框架。
2. 代码下载：下载本项目的源代码。
3. 模型训练：若需使用自定义模型，可通过main.py和main_transformer.py文件中的代码进行训练。
4. 运行程序：
   • 命令行方式：在命令行输入python main.py或python main_transformer.py训练模型，或运行python interactive.py与训练好的模型交互。
   • GUI方式：运行testui.py文件启动图形界面，通过界面与模型交互。
5. 生成藏头诗：输入头韵，程序将生成相应藏头诗并显示。

运行项目前，请确保正确安装所有必要的依赖库，并按需调整代码中的路径和参数设置。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】