文件名称:TransformerCVAE:基于变压器的条件变分自动编码器,可控制故事的产生
文件大小:12.62MB
文件格式:ZIP
更新时间:2024-06-13 07:46:28
Python
变压器CVAE 该存储库包含纸质源代码: @article{fang2021transformer, title={Transformer-based Conditional Variational Autoencoder for Controllable Story Generation}, author={Fang, Le and Zeng, Tao and Liu, Chaochun and Bo, Liefeng and Dong, Wen and Chen, Changyou}, journal={arXiv preprint arXiv:2101.00828}, year={2021} } 获取源数据( , , , )。 数据预处理(数据/)。 训练(从并行性和精度的几种不同实现中选择:train.py,train_dist.py,train_
【文件预览】:
TransformerCVAE-master
----util.py(2KB)
----train.py(36KB)
----eval_ppl_prefix.py(11KB)
----dist_utils.py(6KB)
----generate.py(16KB)
----model.py(26KB)
----train_dist.py(38KB)
----eval_ppl.py(11KB)
----README.md(1KB)
----data()
--------prompt_dataset.py(2KB)
--------util.py(32KB)
--------yelp_dataset.py(757B)
--------__init__.py(0B)
--------plot_dataset.py(1KB)
--------arxiv()
--------arxiv_dataset.py(1KB)
----tsne_plot.py(7KB)
----train_dist_half.py(38KB)
----generate_prefix.py(16KB)