Skip to content

Commit ab1aa5b

Browse files
authored
Merge pull request #88 from yuehhua/doc
Correct VGAE doc
2 parents cb1358f + c7e65c6 commit ab1aa5b

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

docs/src/manual/models.md

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,12 @@ Reference: [Variational Graph Auto-Encoders](https://arxiv.org/abs/1611.07308)
2222
### Variational Graph Autoencoder
2323

2424
```math
25-
X' = \sigma(\hat{D}^{-1/2} \hat{A} \hat{D}^{-1/2} X \Theta)
25+
H = enc(X, A) \\
26+
Z_{\mu}, Z_{logσ} = GCN_{\mu}(H, A), GCN_{\sigma}(H, A) \\
27+
\hat{A} = \sigma (ZZ^T)
2628
```
2729

28-
where ``\hat{A} = A + I``, ``A`` denotes the adjacency matrix, and
29-
``\hat{D} = [\hat{d}_{ij}] = \sum_{j=0} [\hat{a}_{ij}]`` is degree matrix.
30+
where ``A`` denotes the adjacency matrix, ``X`` denotes node features.
3031

3132
```@docs
3233
VGAE
@@ -57,8 +58,8 @@ Reference: [Variational Graph Auto-Encoders](https://arxiv.org/abs/1611.07308)
5758
### Variational Encoder
5859

5960
```math
60-
H = f(X)
61-
μ, logσ = μ(H), Σ(H)
61+
H = enc(X, A) \\
62+
Z_{\mu}, Z_{logσ} = GCN_{\mu}(H, A), GCN_{\sigma}(H, A)
6263
```
6364

6465
```@docs

0 commit comments

Comments
 (0)