【文件属性】:
文件名称:set_transformer:本文'Set Transformer'的TensorFlow实现
文件大小:120KB
文件格式:ZIP
更新时间:2021-05-27 12:22:30
JupyterNotebook
集变器
论文``集变器:基于注意力的置换不变神经网络的框架''的TensorFlow实现
使用Docker(仅限开发人员模式)
在该项目中,添加了Dockerfile和docker-compose.yml文件。 克隆此项目后,可以通过以下步骤使用列出的服务:
docker-compose build
然后使用已在develop模式下安装的此软件包来启动Jupyter笔记本:
docker-compose run -p 8001:8001 jupyter
或开始bash会话:
docker-compose run bash
并执行自动化的单元测试套件:
pytest -W ignore
基本示例用法
from set_transformer . data . simulation import gen_max_dataset
from set_transformer .
【文件预览】:
set_transformer-master
----set_transformer()
--------layers()
--------model.py(578B)
--------__init__.py(0B)
--------data()
--------blocks.py(5KB)
----imgs()
--------transformer.png(50KB)
----Dockerfile(155B)
----requirements.txt(102B)
----.travis.yml(161B)
----setup.py(387B)
----README.md(2KB)
----ipynb()
--------.ipynb_checkpoints()
--------Set_transformer.ipynb(66KB)
----docker-compose.yml(406B)
----tests()
--------test_attention.py(1KB)
--------test_model.py(676B)
--------test_layers.py(420B)
--------__init__.py(0B)
--------test_blocks.py(2KB)
--------test_data_generation.py(447B)
----.gitignore(2KB)