文件名称：*-NeurIPS-2019:NeurIPS 2019论文“通过高效采样从高维事件序列中学习潜在过程”的源代码
文件大小：24.57MB
文件格式：ZIP
更新时间：2024-05-24 09:17:27
Python
*-NeurIPS-2019 NeurIPS 2019论文“通过高效采样从高维事件序列中学习潜在过程”的源代码 环境 的Python 3.5 PyTorch 1.0.1 要求 具有12GB内存的GPU 数据集 使用data_generate.py生成综合数据集 可以从以下位置下载memetracker数据集： ://snap.stanford.edu/data/memetracker9.html 可以从以下下载微博数据集： ： 非常感谢真实数据集的作者。 使用data/#DATA#/preprocess.py预处理下载的数据集，您可以在每个文件夹中获取.pkl文件 要检查每个数据集的统计信息， cck.py在每个文件夹中运行cck.py ，您还可以在其中添加自己的代码行以检查数据集。 快速开始 要训练小型数据集（Syn-Small和Memetracker），您
【文件预览】：
*-NeurIPS-2019-master
----README.md(2KB)
----test.py(3KB)
----data_generate.py(9KB)
----discriminator.py(9KB)
----optimization.py(4KB)
----log()
--------.gitkeep(1B)
----model()
--------main-weibo-512.pt(6.13MB)
--------main-syn-large-512.pt(6.13MB)
--------main-syn-small-512.pt(43KB)
--------main-memetracker-512.pt(31KB)
----train_large.py(8KB)
----model.py(6KB)
----evaluation.py(8KB)
----transformer.py(4KB)
----generator.py(27KB)
----train_small.py(9KB)
----data()
--------memetracker()
--------weibo()
--------syn-small()
--------syn-large()