def __init__(self,adj_train,node_train,size_subgraph,size_root,size_depth):
self.size_root = size_root
self.size_depth = size_depth
size_subgraph = size_root*size_depth
super().__init__(adj_train,node_train,size_subgraph,dict())
self.cy_sampler = cy.RW(self.adj_train.indptr,self.adj_train.indices,self.node_train,\
NUM_PAR_SAMPLER,SAMPLES_PER_PROC,self.size_root,self.size_depth)
def __init__(self, adj_train, node_train, size_subgraph, size_root,
size_depth):
"""
Inputs:
adj_train see super-class
node_train see super-class
size_subgraph see super-class
size_root int, number of root nodes (i.e., number of walkers)
size_depth int, number of hops to take by each walker
Outputs:
None
"""
self.size_root = size_root
self.size_depth = size_depth
size_subgraph = size_root * size_depth
super().__init__(adj_train, node_train, size_subgraph, {})
self.cy_sampler = cy.RW(self.adj_train.indptr, self.adj_train.indices,
self.node_train, NUM_PAR_SAMPLER,
SAMPLES_PER_PROC, self.size_root,
self.size_depth)