• Stars
    star
    138
  • Rank 263,065 (Top 6 %)
  • Language
    Python
  • Created over 4 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

基于Transformer的生成式文本摘要

TransformerSummarization

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

Transformer

使用说明

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

训练

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

简单测试效果

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