Artefacts related to the AISTATS 2021 paper "Learning Temporal Point Processes with Intermittent Observations".
Use a python-3.7 environment and install Tensorflow v1.13.1 and Tensorflow-Probability v0.6.0.
Structure the dataset to separated files for train and test event sequences as below.
dataset_name/event_train.txt dataset_name/time_train.txt dataset_name/event_test.txt dataset_name/time_test.txt
We have also provided a synthetic dataset as a reference.
Use the following command to run IMTPP on the dataset.
python run.py dataset_name
If you use this code in your research, please cite:
@inproceedings{gupta21,
author = {Vinayak Gupta and Abir De and Sourangshu Bhattacharya and Srikanta Bedathur},
title = {Learning Temporal Point Processes with Intermittent Observations},
booktitle = {Proc. of the 24th International Conference on Artificial Intelligence and Statistics (AISTATS)},
year = {2021}
}
In case of any issues, please send a mail to
guptavinayak51 (at) gmail (dot) com