Esempi in Python per new_nbr_req

Esempio n. 1

File: neighbor_sampler.py Progetto: wudidlut/graph-learn

    def _make_req(self, index, src_ids):
        """ Make rpc request.
    """
        if len(self._meta_path) > len(self._expand_factor):
            raise ValueError(
                "input too many meta_path, and can not decide each hops")

        sampler = strategy2op(self._strategy[index], "Sampler")
        req = pywrap.new_nbr_req(self._meta_path[index], sampler,
                                 self._expand_factor[index])
        pywrap.set_nbr_req(req, src_ids)
        return req

Esempio n. 2

File: negative_sampler.py Progetto: zeta1999/graph-learn

 def _make_req(self, ids):
   sampler = strategy2op(self._strategy, "NegativeSampler")
   req = pywrap.new_nbr_req(self._object_type, sampler, self._expand_factor)
   pywrap.set_nbr_req(req, ids)
   return req