Create a SubGraphSampler instance. 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
(self,
graph,
seed_type,
nbr_type,
batch_size,
strategy="random_node")
| 29 | """ |
| 30 | |
| 31 | def __init__(self, |
| 32 | graph, |
| 33 | seed_type, |
| 34 | nbr_type, |
| 35 | batch_size, |
| 36 | strategy="random_node"): |
| 37 | """ Create a SubGraphSampler instance. |
| 38 | Args: |
| 39 | graph (`Graph` object): The graph which sample from. |
| 40 | seed_type (string): Sample seed type, either node type or edge type. |
| 41 | nbr_type (string): Neighbor type of seeds nodes/edges. |
| 42 | batch_size (int): How many nodes will be returned for `get()`. |
| 43 | strategy (string, Optional): Sampling strategy, "random_node" and |
| 44 | "in_order_node" are supported. |
| 45 | """ |
| 46 | self._graph = graph |
| 47 | self._seed_type = seed_type |
| 48 | self._nbr_type = nbr_type |
| 49 | self._batch_size = batch_size |
| 50 | self._strategy = strategy |
| 51 | self._client = self._graph.get_client() |
| 52 | |
| 53 | topology = self._graph.get_topology() |
| 54 | self._node_type = topology.get_src_type(self._nbr_type) |
| 55 | |
| 56 | def get(self): |
| 57 | """ Get sampled `SubGraph`. |
nothing calls this directly
no test coverage detected