I’m interested in doing negative sampling on a graph for node classification. My training graph is a block diagonal matrix where each block is a single training example. I want to perform negative sampling on each block independently – that is, I want to generate negative samples for each block using only the nodes from that same block.
Apart from not calling
dgl.batch(list_of_graphs) and iterating over the individual graphs, how can I go about doing this?