mirrors / dmlc--dgl

项目文件夹

文件

T

Da Zheng 66971c1aa4 [Feature] Add edge sampling for link prediction (#780 )

* add edge sampler.

* add test and run.

* add negative sampling.

* remap the edge subgraph vid.

* negative graph excludes edges of positive edgs.

* remove print.

* avoid sampling NodeFlow when expand_factor or num_hops is 0.

* fix a bug when excluding nodes in negative graph.

* support multigraph.

* exclude positive edges.

* fix memory leak.

* return subgraph object directly.

* fix many problems.

* add comments.

* address comments

2019-08-21 23:05:24 -07:00

5 行

183 B

Python

原始文件 永久链接 Blame 文件历史

 from .sampler import NeighborSampler, LayerSampler, EdgeSampler
 from .randomwalk import *
 from .dis_sampler import SamplerSender, SamplerReceiver
 from .dis_sampler import SamplerPool