上传者: 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