NeighborLoader - Sample neighborhood of selected node ids

[dahe-cvl]

Hi,

Is it possible to sample only selected target nodes with the corresponding neighborhood (k-hop) by using the NeighborLoader class? If yes can you give me an example?

many thanks in advance!

[rusty1s]

You can instantiate your loader just as usual, and then run:

nodes = [0, 1, 2]
out = loader.neighbor_sampler(nodes)
data = loader.transform_fn(out)