Beispiel #1
0
    def __init__(self, adj_train, node_train, size_subgraph):
        """
        Inputs:
            adj_train       see super-class
            node_train      see super-class
            size_subgraph   see super-class

        Outputs:
            None
        """
        self.p_dist = np.zeros(len(node_train))
        super().__init__(adj_train, node_train, size_subgraph, {})
        self.cy_sampler = cy.Node(
            self.adj_train.indptr,
            self.adj_train.indices,
            self.node_train,
            NUM_PAR_SAMPLER,
            SAMPLES_PER_PROC,
            self.p_dist,
            self.size_subgraph,
        )
Beispiel #2
0
 def __init__(self,adj_train,node_train,size_subgraph):
     self.p_dist = np.zeros(len(node_train))
     super().__init__(adj_train,node_train,size_subgraph,dict())
     self.cy_sampler = cy.Node(self.adj_train.indptr,self.adj_train.indices,self.node_train,\
         NUM_PAR_SAMPLER,SAMPLES_PER_PROC,self.p_dist,self.size_subgraph)