MCPcopy Create free account

hub / github.com/GGA23/GrokFormer / functions

Functions25 in github.com/GGA23/GrokFormer

↓ 4 callersFunctionfeature_normalize
(x)
preprocess_node_data.py:92
↓ 3 callersMethod__init__
(self, nclass, nfeat, nlayer=1, hidden_dim=128, dim=32, nheads=1, k=10, tran_dropout=0.0, fea
model.py:90
↓ 3 callersFunctioneig_dgl_adj_sparse
(g, sm=0, lm=0)
preprocess_node_data.py:134
↓ 3 callersFunctioneigen_decompositon
The normalized (unit “length”) eigenvectors,
preprocess_node_data.py:71
↓ 2 callersFunctionnormalize_graph
(g)
preprocess_node_data.py:59
↓ 1 callersFunctiongenerate_node_data
(dataset)
preprocess_node_data.py:190
↓ 1 callersFunctionget_split
(dataset, y, nclass, seed=0)
utils.py:35
↓ 1 callersFunctionload_data
(dataset_str)
preprocess_node_data.py:99
↓ 1 callersFunctionload_fb100_dataset
()
preprocess_node_data.py:164
↓ 1 callersFunctionmain_worker
(args)
train.py:17
↓ 1 callersFunctionparse_index_file
Parse index file.
preprocess_node_data.py:84
↓ 1 callersFunctionseed_everything
(seed)
utils.py:24
↓ 1 callersFunctionset_seed
(seed)
train.py:120
↓ 1 callersMethodtransformer_encoder
(self, h, h_fur)
model.py:132
Method__init__
(self, k, hidden_dim=128)
model.py:13
Method__init__
(self, input_dim, hidden_dim, output_dim)
model.py:37
Method__init__
(self, hidden_size, num_heads, attention_dropout_rate)
model.py:51
Functioncount_parameters
(model)
utils.py:13
Functioneigen_decompositon2
The normalized (unit “length”) eigenvectors,
preprocess_node_data.py:78
Methodforward
(self, e)
model.py:20
Methodforward
(self, x)
model.py:43
Methodforward
(self, q, k, v, attn_bias=None)
model.py:67
Methodforward
(self, e, u, x)
model.py:142
Functiongenerate_signal_data
()
preprocess_node_data.py:21
Functioninit_params
(module)
utils.py:17