Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/cszhangzhen/HGP-SL
/ functions
Functions
27 in github.com/cszhangzhen/HGP-SL
⨍
Functions
27
◇
Types & classes
7
↓ 2 callers
Method
__init__
(self, in_channels, out_channels, cached=False, bias=True, **kwargs)
layers.py:42
↓ 2 callers
Function
compute_test
(loader)
main.py:111
↓ 1 callers
Function
_make_ix_like
(batch)
sparse_softmax.py:53
↓ 1 callers
Function
_threshold_and_support
Sparsemax building block: compute the threshold Args: x: input tensor to apply the sparsemax batch: group indicators Returns:
sparse_softmax.py:61
↓ 1 callers
Method
backward
(ctx, grad_output)
sparse_softmax.py:109
↓ 1 callers
Method
norm
(edge_index, num_nodes, edge_weight, dtype=None)
layers.py:67
↓ 1 callers
Method
norm
(edge_index, num_nodes, edge_weight, dtype=None)
layers.py:117
↓ 1 callers
Method
reset_parameters
(self)
layers.py:62
↓ 1 callers
Function
scatter_sort
(x, batch, fill_value=-1e16)
sparse_softmax.py:27
↓ 1 callers
Function
train
()
main.py:56
Method
__call__
(self, data)
layers.py:15
Method
__init__
(self, improved=False, cached=False, **kwargs)
layers.py:108
Method
__init__
(self, in_channels, ratio=0.8, sample=False, sparse=False, sl=True, lamb=1.0, negative_slop=0.2)
layers.py:157
Method
__init__
(self, args)
models.py:10
Method
__init__
(self)
sparse_softmax.py:125
Method
__repr__
(self)
layers.py:37
Method
__repr__
(self)
layers.py:103
Method
forward
(self, x, edge_index, edge_weight=None)
layers.py:78
Method
forward
(self, x, edge_index, edge_weight)
layers.py:134
Method
forward
(self, x, edge_index, edge_attr, batch=None)
layers.py:173
Method
forward
(self, data)
models.py:34
Method
forward
sparsemax: normalizing sparse transform Parameters: ctx: context object x (Tensor): shape (N, ) batch: gro
sparse_softmax.py:91
Method
forward
(self, x, batch)
sparse_softmax.py:128
Method
message
(self, x_j, norm)
layers.py:95
Method
message
(self, x_j, norm)
layers.py:149
Method
update
(self, aggr_out)
layers.py:98
Method
update
(self, aggr_out)
layers.py:152