【文件属性】：
文件名称：MemN2N-tensorflow：Tensorflow中的“端到端内存网络”
文件大小：1.9MB
文件格式：ZIP
更新时间：2021-02-03 15:22:17
nlp tensorflow memory-network NaturallanguageprocessingPython
Tensorflow中的端到端内存网络
Tensorflow实现用于语言建模（请参阅第5节）。 来自Facebook的原始割炬代码可在找到。
先决条件
此代码需要 。 data目录中有一组样本Penn Tree Bank（PTB）语料库，这是衡量这些模型质量的流行基准。 但是您可以使用自己的文本数据集，其格式应如下。
当您使用docker image tensorflw / tensorflow：latest-gpu时，您需要将来的python包。
$ pip install future
如果要使用--show True选项，则需要安装python软件包progress 。
$ pip install progress
用法
要训练具有6个跃点且内存大小为100的模型，请运行以下命令：
$ python main.py --nhop 6 --mem_size 100
要查看所有培训选项，请运行：
$ python main.py --help
它将打印：
usage: main.py [-h] [--edim EDIM] [--lindim LINDIM] [--nh
【文件预览】：
MemN2N-tensorflow-master
----data.py(861B)
----utils.py(140B)
----main.py(2KB)
----model.py(9KB)
----LICENSE(1KB)
----README.md(5KB)
----data()
--------ptb.test.txt(439KB)
--------ptb.train.txt(4.87MB)
--------ptb.valid.txt(390KB)
----.gitignore(946B)