MCPcopy Index your code

hub / github.com/EdisonLeeeee/SpikeGCL / functions

Functions42 in github.com/EdisonLeeeee/SpikeGCL

⨍ Functions 42◇ Types & classes 11
↓ 3 callersMethodencode
(self, x, edge_index, edge_weight=None)
spikegcl/model.py:89
↓ 3 callersFunctiongaussian
Gaussian PDF with broadcasting.
spikegcl/neuron.py:17
↓ 3 callersMethodreset
(self, reset_type: str = "subtract")
spikegcl/neuron.py:161
↓ 2 callersMethod__init__
( self, v_threshold=1.0, v_reset=0.0, alpha=1.0, surrogate="triangle", detach=True, )
spikegcl/neuron.py:149
↓ 2 callersMethodbackward
(ctx, grad_output)
spikegcl/neuron.py:37
↓ 2 callersMethoddecode
(self, spikes)
spikegcl/model.py:107
↓ 2 callersFunctioneval_acc
(model, x, y)
spikegcl/evaluate.py:16
↓ 1 callersFunctioncreat_activation_layer
(activation)
spikegcl/model.py:9
↓ 1 callersFunctioncreat_snn_layer
( alpha=2.0, surrogate="sigmoid", v_threshold=5e-3, snn="PLIF", )
spikegcl/model.py:20
↓ 1 callersFunctionget_dataset
(root, dataset, num_val=0.1, num_test=0.8)
spikegcl/dataset.py:7
↓ 1 callersMethodloss
(self, postive, negative, margin=0.0)
spikegcl/model.py:133
↓ 1 callersFunctionnode_cls_downstream_task_eval
(input_emb, data, num_classes, lr, wd, cls_epochs=100,
spikegcl/evaluate.py:67
↓ 1 callersFunctionread_parser
()
main.py:13
↓ 1 callersMethodreset
(self, reset_type: str = "subtract")
spikegcl/neuron.py:203
↓ 1 callersMethodreset_parameters
(self)
spikegcl/evaluate.py:34
↓ 1 callersFunctiontab_printer
Function to print the logs in a nice tabular format. Note ---- Package `Texttable` is required. Run `pip install Texttable` if was no
spikegcl/utils.py:4
↓ 1 callersFunctiontest
(embeds, data, num_classes)
spikegcl/evaluate.py:6
↓ 1 callersFunctiontrain
()
main.py:154
↓ 1 callersFunctiontrain_cls
(cls, x, y, train_mask, val_mask, test_mask, lr=1e-2, weight_decay=1e-5, epochs=100)
spikegcl/evaluate.py:38
Method__init__
( self, tau=1.0, v_threshold=1.0, v_reset=0.0, alpha=1.0,
spikegcl/neuron.py:184
Method__init__
( self, tau=1.0, v_threshold=1.0, v_reset=0.0, alpha=1.0,
spikegcl/neuron.py:226
Method__init__
(self, in_dim, out_dim)
spikegcl/evaluate.py:26
Method__init__
( self, in_channels: int, hidden_channels: int, out_channels: int, T:
spikegcl/model.py:43
Functionarctanspike
(x, thresh=torch.tensor(1.0), alpha=torch.tensor(10.0))
spikegcl/neuron.py:135
Methodbackward
(ctx, grad_output)
spikegcl/neuron.py:52
Methodbackward
(ctx, grad_output)
spikegcl/neuron.py:69
Methodbackward
(ctx, grad_output)
spikegcl/neuron.py:88
Methodbackward
(ctx, grad_output)
spikegcl/neuron.py:102
Methodbackward
(ctx, grad_output)
spikegcl/neuron.py:111
Methodforward
(ctx, x, alpha)
spikegcl/neuron.py:32
Methodforward
(self, dv)
spikegcl/neuron.py:168
Methodforward
(self, dv)
spikegcl/neuron.py:210
Methodforward
(self, dv)
spikegcl/neuron.py:253
Methodforward
(self, x)
spikegcl/evaluate.py:30
Methodforward
(self, x, edge_index, edge_weight=None)
spikegcl/model.py:113
Functionheaviside
(x: torch.Tensor)
spikegcl/neuron.py:13
Functionmgspike
(x, thresh=torch.tensor(1.0), alpha=torch.tensor(0.5))
spikegcl/neuron.py:123
Methodreset
(self, reset_type: str = "subtract")
spikegcl/neuron.py:246
Functionreset_net
(net: nn.Module, reset_type: str = "subtract")
spikegcl/neuron.py:7
Functionsigmoidspike
(x, thresh=torch.tensor(1.0), alpha=torch.tensor(1.0))
spikegcl/neuron.py:127
Functionsuperspike
(x, thresh=torch.tensor(1.0), alpha=torch.tensor(10.0))
spikegcl/neuron.py:119
Functiontrianglespike
(x, thresh=torch.tensor(1.0), alpha=torch.tensor(1.0))
spikegcl/neuron.py:131