MCPcopy Create free account

hub / github.com/BorealisAI/SLAPS-GNN / functions

Functions52 in github.com/BorealisAI/SLAPS-GNN

↓ 6 callersFunctionsample_mask
Create mask.
citation_networks.py:42
↓ 4 callersFunctionapply_non_linearity
(tensor, non_linearity, i)
utils.py:16
↓ 4 callersFunctionget_random_mask
(features, r, nr)
utils.py:27
↓ 3 callersMethodget_loss_fixed_adj
(self, model, mask, features, labels)
main.py:32
↓ 3 callersMethodget_loss_learnable_adj
(self, model, mask, features, labels, Adj)
main.py:25
↓ 3 callersMethodget_loss_masked_features
(self, model, features, mask, ogb, noise, loss_t)
main.py:47
↓ 3 callersMethodhalf_val_as_train
(self, val_mask, train_mask)
main.py:70
↓ 3 callersFunctionload_data
(args)
data_loader.py:41
↓ 3 callersFunctionnearest_neighbors
(X, k, metric)
utils.py:49
↓ 3 callersMethodprint_results
(self, validation_accu, test_accu)
main.py:312
↓ 2 callersMethod__init__
(self, input_size)
layers.py:44
↓ 2 callersMethod__init__
(self, nlayers, isize, hsize, osize, mlp_epochs, k, knn_metric, non_linearity, i, sparse, mlp_act)
graph_generator.py:102
↓ 2 callersMethod__init__
(self, in_channels, hidden_channels, out_channels, num_layers, dropout, dropout_adj, Adj, sparse)
model.py:78
↓ 2 callersFunctionaccuracy
(preds, labels)
utils.py:44
↓ 2 callersFunctioncal_similarity_graph
(node_embeddings)
utils.py:112
↓ 2 callersFunctionget_random_mask_ogb
(features, r)
utils.py:38
↓ 2 callersMethodinternal_forward
(self, h)
graph_generator.py:71
↓ 2 callersMethodinternal_forward
(self, h)
graph_generator.py:125
↓ 2 callersFunctionknn_fast
(X, k, b)
utils.py:128
↓ 2 callersFunctionnormalize
(adj, mode, sparse=False)
utils.py:82
↓ 2 callersFunctiontop_k
(raw_graph, K)
utils.py:117
↓ 2 callersMethodtrain_classification_gcn
(self, Adj, features, nfeats, labels, nclasses, train_mask, val_mask, test_mask, args)
main.py:80
↓ 1 callersMethodforward
(self, features)
graph_generator.py:154
↓ 1 callersMethodget_adj
(self, h)
model.py:55
↓ 1 callersFunctionload_citation_network
(dataset_str)
citation_networks.py:49
↓ 1 callersFunctionload_ogb_data
(dataset_str)
data_loader.py:16
↓ 1 callersMethodmlp_knn_init
(self)
graph_generator.py:135
↓ 1 callersFunctionnearest_neighbors_pre_elu
(X, k, metric, i)
utils.py:74
↓ 1 callersFunctionnearest_neighbors_pre_exp
(X, k, metric, i)
utils.py:66
↓ 1 callersFunctionparse_index_file
Parse index file.
citation_networks.py:34
↓ 1 callersFunctionsymmetrize
(adj)
utils.py:108
↓ 1 callersMethodtrain_end_to_end
(self, args)
main.py:217
↓ 1 callersMethodtrain_knn_gcn
(self, args)
main.py:129
↓ 1 callersMethodtrain_two_steps
(self, args)
main.py:151
Method__init__
(self, input_size, output_size)
layers.py:15
Method__init__
(self, input_size, output_size)
layers.py:32
Method__init__
(self, features, non_linearity, k, knn_metric, i, sparse)
graph_generator.py:15
Method__init__
(self, nlayers, isize, k, knn_metric, non_linearity, i, sparse, mlp_act)
graph_generator.py:58
Method__init__
(self)
main.py:22
Method__init__
(self, nlayers, in_dim, hidden_dim, nclasses, dropout, dropout_adj, features, k, knn_metric, i_,
model.py:14
Method__init__
(self, in_channels, hidden_channels, out_channels, num_layers, dropout, dropout_adj, sparse)
model.py:117
Methodforward
(self, input, A, sparse=False)
layers.py:22
Methodforward
(self, x, g)
layers.py:36
Methodforward
(self, input)
layers.py:49
Methodforward
(self, h)
graph_generator.py:35
Methodforward
(self, features)
graph_generator.py:81
Methodforward
(self, features, x)
model.py:62
Methodforward
(self, x)
model.py:100
Methodforward
(self, x, adj_t)
model.py:138
Methodget_loss_adj
(self, model, features, feat_ind)
main.py:39
Methodinit_para
(self)
layers.py:19
Functionnearest_neighbors_sparse
(X, k, metric)
utils.py:56