Skip to content

Commit

Permalink
Fix typo in SAGPool (#5244)
Browse files Browse the repository at this point in the history
* Fix typo in SAGPool

Removed a piece of text belonging to the top-k operator.

* Update torch_geometric/nn/pool/sag_pool.py

Co-authored-by: Padarn Wilson <padarn.wilson@grabtaxi.com>

Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>
Co-authored-by: Padarn Wilson <padarn.wilson@grabtaxi.com>
  • Loading branch information
3 people authored Aug 20, 2022
1 parent fd77f03 commit 39c7e88
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions torch_geometric/nn/pool/sag_pool.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,10 +34,8 @@ class SAGPooling(torch.nn.Module):
\mathbf{X}^{\prime} &= (\mathbf{X} \odot \mathbf{y})_{\mathbf{i}}
\mathbf{A}^{\prime} &= \mathbf{A}_{\mathbf{i},\mathbf{i}},
\mathbf{A}^{\prime} &= \mathbf{A}_{\mathbf{i},\mathbf{i}}.
where nodes are dropped based on a learnable projection score
:math:`\mathbf{p}`.
Projections scores are learned based on a graph neural network layer.
Args:
Expand Down

0 comments on commit 39c7e88

Please sign in to comment.