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)