search

Graph-Machine-Learning-Group / grin

Official repository for the paper "Filling the G_ap_s: Multivariate Time Series Imputation by Graph Neural Networks" (ICLR 2022)
https://arxiv.org/abs/2108.00298
149 stars 30 forks source link

Testing GRIN on a new dataset #7

Closed RoufaidaLaidi closed 2 years ago

RoufaidaLaidi commented 2 years ago

Hello, Thanks for sharing the code for the paper. I would like to test GRIN for imputation on my own dataset. The dataset is presented in a csv file and I create the mask that simulates the missing data using a random function. Can you indicate to me how to proceed to include my own dataset? Thanks in advance.

andreacini commented 2 years ago

Hi! Sorry for the late reply, you should wrap your dataset in a class similar to those in lib/datasets.

I also suggest you check our library, where you'll find updated code for GRIN and other methods.