NeighborSampler parameter: num_hops

Is the value of num_hops in calling NeighborSampler the same as the value of NodeFlow.num_layers that it returns?

There are two different ways of doing this

You are right when you want to use FullNeighborSampler
sampler = dgl.dataloading.MultiLayerFullNeighborSampler(nof_layers)

You can also set the number of neighbors you want to sample for each layer like below
sampler = dgl.dataloading.MultiLayerNeighborSampler([15, 10]) #for 2 layers

See the documentation for more details: dgl.dataloading — DGL 0.6.1 documentation

1 Like