search

rusty1s / pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch
https://arxiv.org/abs/1711.08920
MIT License
172 stars 36 forks source link

graph preprocessing for SplineConv layer #20

Closed Original-Thunderbird closed 3 years ago

Original-Thunderbird commented 3 years ago

Hello, I am not quite sure how graphs shall be preprocessed for the SplineConv layer. My dataset contains graphs that each with different number of vertices, can they be directly fed into the layer after padding or they have to be preprocessed so that they have the same number of vertices? If only padding is needed, shall I do padding across the whole dataset or within batches? Also, may I ask if there is any blogs/tutorials you recommend for graph preprocessing in spatial graph convolution? Thanks a lot in advance!

rusty1s commented 3 years ago

Have you looked at PyTorch Geometric? There is a basic introduction about the graph data structure, as well as some examples of using SplineConv.

Original-Thunderbird commented 3 years ago

Many thanks!