shchur / ifl-tpp

Implementation of "Intensity-Free Learning of Temporal Point Processes" (Spotlight @ ICLR 2020)
https://openreview.net/forum?id=HygOjhEYDH
MIT License
81 stars 32 forks source link

Implementation on missing data imputation #18

Closed yugongg closed 2 years ago

yugongg commented 2 years ago

Hi @shchur, thanks for releasing the code! I'm wondering if it is possible that you could provide the code for the section of missing data imputation - Sec. 5.4 & F.4 MISSING DATA IMPUTATION from your paper.

I'm curious about your implementation of feeding imputations to the RNN. If keep your current framework, to include the imputations to be the history, the batch will keep changing and we need get_features(batch) & get_context(features) every time we have a new imputed event. This gives a very slow training process. Could you provide your implementation on this part, or give me some suggestions on implementing this 'training while imputing'? Thanks!

shchur commented 2 years ago

Hi @yugongg. We looked for the code for this particular experiment but, unfortunately, it seems like we lost it somewhere between refactoring & moving from the internal repo to Github, I'm sorry about that :(

I can recommend having a look at this paper https://proceedings.mlr.press/v130/gupta21a.html that studies training of neural TPPs with automatically imputing the missing data in more detail. Here's the link to their code https://github.com/data-iitd/imtpp. I hope this helps.