-
使用Tensorflow实现seq2seq的文本摘要:text-summarization-tensorflow
资源介绍
张量流文本摘要
使用的文本汇总的简单Tensorflow实现。
模型
具有注意力机制的编码器-解码器模型。
词嵌入
使用初始化单词嵌入。
编码器
使用了带有LSTM单元。
解码器
使用LSTM进行训练,并使用进行推理。
注意机制
使用进行体重归一化。
要求
Python 3
Tensorflow(> = 1.8.0)
点安装-r requirements.txt
用法
准备数据
数据集位于 。 在项目根目录中找到summary.tar.gz文件。 然后,
$ python prep_data.py
要使用Glove预训练的嵌入,请通过下载
$ python prep_data.py --