"...en/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a39d42b91dd561d4487240e59455d9e2a53654fb"
Unverified Commit adea4ba1 authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

fix doc. (#1100)

parent 869093e8
...@@ -493,21 +493,23 @@ class EdgeSampler(object): ...@@ -493,21 +493,23 @@ class EdgeSampler(object):
When negative edges are created, a batch of negative edges are also placed When negative edges are created, a batch of negative edges are also placed
in a subgraph. in a subgraph.
Currently, negative_mode only supports: Currently, `negative_mode` only supports:
'head': the negative edges are generated by corrupting head nodes
with uniformly randomly sampled nodes, * 'head': the negative edges are generated by corrupting head nodes with uniformly randomly sampled nodes,
'tail': the negative edges are generated by corrupting tail nodes
with uniformly randomly sampled nodes, * 'tail': the negative edges are generated by corrupting tail nodes with uniformly randomly sampled nodes,
'PBG-head': the negative edges are generated by corrupting a set
of head nodes with the same set of nodes uniformly randomly sampled * 'PBG-head': the negative edges are generated by corrupting a set \
of head nodes with the same set of nodes uniformly randomly sampled \
from the graph. Please see Pytorch-BigGraph for more details. from the graph. Please see Pytorch-BigGraph for more details.
'PBG-tail': the negative edges are generated by corrupting a set
* 'PBG-tail': the negative edges are generated by corrupting a set \
of tail nodes with the same set of nodes similar to 'PBG-head'. of tail nodes with the same set of nodes similar to 'PBG-head'.
The sampler returns EdgeSubgraph, where a user can access the unique head nodes The sampler returns EdgeSubgraph, where a user can access the unique head nodes
and tail nodes directly. and tail nodes directly.
When the flag return_false_neg is turned on, the sampler will also check When the flag `return_false_neg` is turned on, the sampler will also check
if the generated negative edges are true negative edges and will return if the generated negative edges are true negative edges and will return
a vector that indicates false negative edges. The vector is stored in a vector that indicates false negative edges. The vector is stored in
the negative graph as `false_neg` edge data. the negative graph as `false_neg` edge data.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment