K-fold cross-validation using PyTorch Geometric

[bianchiniriccardo95]

Hello!
First of all, thanks for this amazing tool.
I have a question:
I want to perform node classification on my network composed of 2410 nodes. Since now, I split my nodes using the function AddTrainValTestMask function in order to validate the machine, but my idea is to perform a K-fold cross validation of the network, to gain better confidence about the goodness of the GCN.
How can I do this? How can I split my nodes in the different masks multiple times? Is there a specific function?
Thanks in advance.

[rusty1s]

Such a function doesn't exist yet, I'm sorry. I suggest to use sklearn.model_selection.StratifiedKFold for this. Here is an example of it within PyG.

[bianchiniriccardo95]

Thank you very much for your help!

[WhItEG-bit]

Thanks，help a lot!