Can NeighborSampler support weights ?

[xiachenrui]

Hi, all:

Can we sample the graph with (edge) weight when using NeighborSampler ? This strategy is avaliable in DGL (https://docs.dgl.ai/en/0.8.x/generated/dgl.sampling.sample_neighbors.html#dgl.sampling.sample_neighbors)

Best !
Chen-Rui

[wsad1]

NeighborSampler has been depreciated in favour of NeighborLoader. And NeighborLoader currently doesn't support defining a weight/prob for each edge which can be used while sampling.
But this a useful feature to add @mananshah99 , @rusty1s.

[xiachenrui]

Thanks for your kind reply !

[rusty1s]

Yes, we have this on our agenda with @pyg-team/intel-team :)