search

tjiagoM / spatio-temporal-brain

A Deep Graph Neural Network Architecture for Modelling Spatio-temporal Dynamics in rs-fMRI Data
Apache License 2.0
43 stars 8 forks source link

help #18

Closed HarlanZhou closed 2 years ago

HarlanZhou commented 3 years ago

how to create my custiom dataset, if i use this model to training

tjiagoM commented 3 years ago

Hi!

You can see the code I created to load data from the uk biobank and human connectome project (HCP) here: https://github.com/tjiagoM/spatio-temporal-brain/blob/master/datasets.py

You will probably need to know how data loading works with pytorch geometric (https://pytorch-geometric.readthedocs.io/en/latest/notes/create_dataset.html), before looking to this module.

Hope this helps!

HarlanZhou commented 3 years ago

Hi!

You can see the code I created to load data from the uk biobank and human connectome project (HCP) here: https://github.com/tjiagoM/spatio-temporal-brain/blob/master/datasets.py

You will probably need to know how data loading works with pytorch geometric (https://pytorch-geometric.readthedocs.io/en/latest/notes/create_dataset.html), before looking to this module.

Hope this helps!

Could this model be used in another way to build data sets from documents

from torch_geometric.data import Data, DataLoader

data_list = [Data(...), ..., Data(...)]
loader = DataLoader(data_list, batch_size=32)
tjiagoM commented 3 years ago

Hi again! Sorry, but I'm not sure I follow your question and how the code relates to it.