NeighborSampler depth

Hi everyone,

I want to sample with data loader neighbors up to a given depth like so (K=2):
Screenshot 2022-09-21 at 12.59.17

How can I achieve this? NeighborSampler from what I understand you can only specify the number of neighbors per GNN layer, but not the neighbourhood depth.

You can specify a list of fanouts for depth more than one. For instance,

NeighborSampler([5, 10])

gives you two hops. The first layer (the second hop) is 5 neighbors and the second layer (the first hop) 10.

2 Likes

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.