Go to file

huaian_zhou 3a6b17be2d 添加教程文件		2024-01-13 14:59:53 +08:00
doc	添加教程文件	2024-01-13 14:59:53 +08:00
model	修改项目文件结构	2024-01-13 14:50:44 +08:00
trans_classification	基于Transformer架构的对联生成模型	2024-01-13 14:51:06 +08:00
trans_couplet	基于Transformer架构的对联生成模型	2024-01-13 14:51:06 +08:00
trans_translation	修改README	2024-01-13 14:51:13 +08:00
README.md	修改README	2024-01-13 14:51:13 +08:00

README.md

Transformer模型项目

基于Transformer结构的文本翻译模型

环境准备
- Python 3.9
- Pytorch 1.12
- torchtext 0.13.0
- 安装spaCy分词器
  - pip install de_core_news_sm-3.7.0.tar.gz
  - pip install en_core_web_sm-3.7.1.tar.gz
模型训练
- python train.py
文本翻译
- python translate.py

基于Transformer Encoder的文本分类模型

模型训练
- python train.py
文本分类
- python classification.py

基于Transformer结构的对联生成模型

模型训练
- python train.py
生成对联
- python couplet.py