Skip to content

HorieYuan/TransformerTextSummarization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TransformerSummarization

基于Transformer的生成式文本摘要模型,基于Mxnet/gluon。

Transformer

使用说明

大部分超参数可以在 hyper_parameters.py 文件中进行设置。

训练

设置完参数之后,运行 train.py 文件。

简单测试效果

运行 summarize.py 文件,按照提示输入原文,可根据训练结果调整测试时加载的模型轮次。

About

基于Transformer的生成式文本摘要

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages