Transformer-Hawkes-Process:变压器霍克斯过程代码,ICML 2020-源码

上传者: 42150341 | 上传时间: 2021-12-04 16:48:40 | 文件大小: 12KB | 文件类型: -
变形金刚霍克斯过程 源代码。 运行代码 依存关系 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

文件下载

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明