cugraph_pyg.sampler.cugraph_sampler.EXPERIMENTAL__CuGraphSampler#

class cugraph_pyg.sampler.cugraph_sampler.EXPERIMENTAL__CuGraphSampler(data: Tuple[CuGraphStore, CuGraphStore], method: str = 'uniform_neighbor', **kwargs)[source]#

Duck-typed version of PyG’s BaseSampler

Attributes:

edge_permutation
method

Methods

sample_from_nodes(sampler_input)

Sample nodes using this CuGraphSampler's sampling method (which is set at initialization) and the input node data passed to this function.

sample_from_edges

__init__(data: Tuple[CuGraphStore, CuGraphStore], method: str = 'uniform_neighbor', **kwargs)[source]#

Methods

`__init__`(data[, method])
`sample_from_edges`(index)
`sample_from_nodes`(sampler_input)	Sample nodes using this CuGraphSampler's sampling method (which is set at initialization) and the input node data passed to this function.

Attributes

`SAMPLING_METHODS`
`UNIFORM_NEIGHBOR`
`edge_permutation`
`method`