MCPcopy Create free account
hub / github.com/alibaba/graph-learn / subgraph_sampler

Method subgraph_sampler

graphlearn/python/graph.py:1005–1027  ·  view source on GitHub ↗

Sampler for sample SubGraph. Args: graph (`Graph` object): The graph which sample from. seed_type (string): Sample seed type, either node type or edge type. nbr_type (string): Neighbor type of seeds nodes/edges. batch_size (int): How many nodes will be returned for `get(

(self,
                       seed_type,
                       nbr_type,
                       batch_size=64,
                       strategy="random_node")

Source from the content-addressed store, hash-verified

1003 return knn.search(node_type, inputs, option.k)
1004
1005 def subgraph_sampler(self,
1006 seed_type,
1007 nbr_type,
1008 batch_size=64,
1009 strategy="random_node"):
1010 """Sampler for sample SubGraph.
1011
1012 Args:
1013 graph (`Graph` object): The graph which sample from.
1014 seed_type (string): Sample seed type, either node type or edge type.
1015 nbr_type (string): Neighbor type of seeds nodes/edges.
1016 batch_size (int): How many nodes will be returned for `get()`.
1017 strategy (string, Optional): Sampling strategy. "random_node".
1018 Return:
1019 An `SubGraphSampler` object.
1020
1021 """
1022 sampler = utils.strategy2op(strategy, "SubGraphSampler")
1023 return getattr(samplers, sampler)(self,
1024 seed_type,
1025 nbr_type,
1026 batch_size=batch_size,
1027 strategy=strategy)
1028
1029 def get_stats(self):
1030 req = pywrap.new_get_stats_request()

Callers 2

Calls

no outgoing calls

Tested by 2