search

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
MIT License
2.66k stars 377 forks source link

Custom Dataset Question #151

Closed GriffinBonner closed 2 years ago

GriffinBonner commented 2 years ago

I am trying to create a dataset which represents a static graph and temporal signal. I understand that the "Temporal Data Snapshots" are pytorch geometric "Data" objects, and I understand how to create them. However, I do not understand how to interface with the "Temporal Signal Iterators" ie., "StaticGraphTemporalSignal." Do I create a list of "Data" objects and then pass that to the "StaticGraphTemporalSignal" constructor? The documentation seems to suggest that several of these structures all take the same input arguments of the parameters of a "Data" object in their constructor, I just am having difficulty understanding how to specify the time varying node features of my custom dataset. Can anyone provide a tutorial or example code for creating a custom pytorch geometric temporal dataset? I haven't found this anywhere. Thank you.

benedekrozemberczki commented 2 years ago

Is not this something that you could do based on the existing benchmark/toy datasets?