# poet

**Repository Path**: gystar/poet

## Basic Information

- **Project Name**: poet
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2020-11-20
- **Last Updated**: 2021-01-06

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

LSTM写诗实现
模型：根据前面所有字，预测下一个最大可能出现的字。
训练：先使用全诗集来进行训练模型

依赖：
pytorch(1.7)
flask

使用方法：
运行server.py,在网页中打开对应的页面 

有三种模式：
1.根据第一句（或者说前几个字）
2.藏头诗
3.根据关键字

因为model比较大，没有上传到代码库。
训练非常慢，建议使用GPU进行训练，get_model会加载存在的数据文件，指定continue_train=true可以进行继续训练
build目录中有个训练好的模型model_best.pkl，用的时候将其改名为model.pkl即可自动加载