dgl.graphbolt.exclude_seed_edgesΒΆ

dgl.graphbolt.exclude_seed_edges(minibatch: dgl.graphbolt.minibatch.MiniBatch, include_reverse_edges: bool = False, reverse_etypes_mapping: Optional[Dict[str, str]] = None)[source]ΒΆ

Exclude seed edges with or without their reverse edges from the sampled subgraphs in the minibatch.

Parameters

  • minibatch (MiniBatch) – The minibatch.

  • reverse_etypes_mapping (Dict[str, str] = None) – The mapping from the original edge types to their reverse edge types.