文件名称：Transformer-Hawkes-Process:变压器霍克斯过程代码，ICML 2020
文件大小：12KB
文件格式：ZIP
更新时间：2024-05-29 17:07:01
Python
变形金刚霍克斯过程 源代码。 运行代码 依存关系 Python 3.7。 包含所有必需的软件包。 版本1.4.0。 指示 将数据文件夹放在根文件夹中，并相应地修改run.sh中的数据条目。 数据集可。 bash run.sh运行代码。 笔记 目前，该代码仅支持单个GPU训练，但是支持多个GPU的扩展应该很容易。 所报告的事件时间预测RMSE和数据集中提供的时间戳不是同一单位，即，所提供的时间戳可以以分钟为单位，但是报告的结果以小时为单位。 除了run.sh中的因素外，还有几个可以更改的因素： 在Main.py中的函数train_epoch中，需要适当缩放事件时间预测平方误差以稳定训练。 同时，还要在Utils.py中的time_loss函数中缩放diff变量。 在Utils.py函数log_likelihood中，用户可以选择使用数值积分还是蒙特卡洛积分。 在Transfo
【文件预览】：
Transformer-Hawkes-Process-master
----preprocess()
--------Dataset.py(2KB)
----Utils.py(4KB)
----run.sh(437B)
----README.md(2KB)
----Main.py(9KB)
----transformer()
--------Layers.py(924B)
--------Modules.py(638B)
--------Constants.py(9B)
--------Models.py(6KB)
--------SubLayers.py(3KB)
----.gitignore(33B)