blank_language_model:空白语言模型

时间:2021-05-31 05:38:14
【文件属性】:
文件名称:blank_language_model:空白语言模型
文件大小:176KB
文件格式:ZIP
更新时间:2021-05-31 05:38:14
Python 空白语言模型 此存储库包含我们的 EMNLP 2020 论文的代码: Tianxiao Shen*、Victor Quach*、Regina Barzilay 和 Tommi Jaakkola(*:同等贡献) 给定带有一个或多个空白的部分指定文本,BLM 将使用与上下文一致的可变数量的标记填充空白,使其成为文本编辑和重写的理想选择。 输入:他们也有 ___ 其中 ___ 。 输出:他们也有冰淇淋哪一个真的很好. 演示 我们有一个使用构建的在线演示,可 或者通过运行在本地尝试: streamlit run app.py 依赖关系 我们的代码基于框架。 已在 PyTorch 1.6.0、PyTorch Lightning 1.0.7 中测试 下载资料 运行以下命令下载处理后的 Yelp 和 Yahoo 数据集: bash download_data.sh 训练 要在 Yelp 否
【文件预览】:
blank_language_model-release
----.gitignore(1KB)
----get_model.sh(190B)
----models()
--------blm.py(5KB)
--------get_canvas.cpp(3KB)
--------__init__.py(324B)
--------lm.py(3KB)
--------lblm.py(6KB)
--------inst.py(6KB)
--------torch_utils.py(4KB)
----LICENSE(11KB)
----download_data.sh(189B)
----utils.py(2KB)
----README.md(4KB)
----test.py(5KB)
----dataset.py(2KB)
----optimizer.py(757B)
----app.py(2KB)
----vocab.py(1KB)
----img()
--------model.png(162KB)
----transformer()
--------Beam.py(3KB)
--------Models.py(7KB)
--------Layers.py(2KB)
--------__init__.py(409B)
--------SubLayers.py(3KB)
--------Translator.py(7KB)
--------Constants.py(113B)
--------Optim.py(2KB)
--------Modules.py(729B)
----train.py(7KB)

网友评论