Method __init__

tf_euler/python/dataflow/whole_dataflow.py:27–35

(self, metapath,
                 add_self_loops=True,
                 **kwargs)

class WholeDataFlow(NeighborDataFlow):
    def __init__(self, metapath,
                 add_self_loops=True,
                 **kwargs):
        super(WholeDataFlow, self).__init__(num_hops=len(metapath),
                                            add_self_loops=add_self_loops)
        self.neighbor_type = metapath[0]
        for n_type in metapath:
            if not n_type == self.neighbor_type:
                raise ValueError('Metapath should be the same in whole graph sampler.')

    def get_self_neighbors(self, n_id):
        edge_src = []