MCPcopy Create free account

hub / github.com/NineAbyss/ZeroG / functions

Functions109 in github.com/NineAbyss/ZeroG

↓ 5 callersMethod__init__
(self, in_dim, emb_dim, num_layer, kernel='gcn', drop_ratio=0, act='relu', norm='batchnorm',
code/model.py:49
↓ 4 callersMethod__init__
(self, root, name, transform=None, pre_transform=None)
code/dataset_benchmark.py:209
↓ 3 callersFunctionload_dataset
Wrapper functions for data loader
code/dataset_benchmark.py:624
↓ 3 callersFunctionobtain_act
Return activation function module
code/utils.py:183
↓ 2 callersMethod__init__
(self, input_dim, hidden_dim, output_dim)
code/utils.py:281
↓ 2 callersMethoddiscriminate
(self, h, summary)
code/model.py:183
↓ 2 callersFunctioneval
(i,idx,model, test_data, args)
code/main_TextBP_benchmark.py:166
↓ 2 callersMethodget
(self, idx)
code/SubgraphDataset.py:98
↓ 2 callersMethodget_taxonomy
(self, path)
code/dataset_benchmark.py:367
↓ 2 callersMethodnormalize_adjacency_matrix
(self, edge_index, num_nodes)
code/st_model.py:117
↓ 2 callersMethodnormalize_adjacency_matrix
(self, edge_index, num_nodes)
code/st_model.py:297
↓ 1 callersMethod__init__
(self, args)
code/st_model.py:11
↓ 1 callersMethod__init__
(self, data, num_hops=1, max_nodes=100, dataset_name="Cora", transform=None, pre_transform=None)
code/SubgraphDataset.py:52
↓ 1 callersMethod_create_subgraphs
(self)
code/SubgraphDataset.py:72
↓ 1 callersMethod_find_valid_subgraphs
(self)
code/SubgraphDataset.py:167
↓ 1 callersMethod_random_split
split the dataset into training set, validation set and testing set
code/dataset_benchmark.py:167
↓ 1 callersFunctionbuild_args
()
code/main_TextBP_benchmark.py:74
↓ 1 callersMethodencding_mask
(self, x)
code/model.py:238
↓ 1 callersMethodforward
(self, x, edge_index, edge_weigt=None, batch=None)
code/model.py:225
↓ 1 callersMethodget
(self, idx)
code/SubgraphDataset.py:188
↓ 1 callersMethodget_texts
(self, root)
code/dataset_benchmark.py:345
↓ 1 callersFunctioninfomax_corruption
(features, batch)
code/utils.py:237
↓ 1 callersMethodlabel_emb_func
(self, root)
code/dataset_benchmark.py:430
↓ 1 callersMethodlabel_emb_func
(self, root)
code/dataset_benchmark.py:490
↓ 1 callersFunctionload_link_dataset
(root, name, args)
code/dataset_benchmark.py:610
↓ 1 callersFunctionload_node_dataset
Wrapper functions for node level data loader
code/dataset_benchmark.py:574
↓ 1 callersFunctionload_tu_dataset
Data loader for TUDataset
code/dataset_benchmark.py:313
↓ 1 callersFunctionlogger_config
(log_path, logging_name)
code/main_TextBP_benchmark.py:21
↓ 1 callersFunctionobtain_norm
Return normalization function module
code/utils.py:205
↓ 1 callersFunctionsce_loss
(x, y, alpha=3)
code/utils.py:424
↓ 1 callersFunctionset_random_seed
(seed)
code/utils.py:24
↓ 1 callersMethodtext_emb
(self)
code/dataset_benchmark.py:420
↓ 1 callersMethodtext_emb
(self)
code/dataset_benchmark.py:536
↓ 1 callersMethodzero_shot_eval
(self, node_embeds, label_embeds, data)
code/st_model.py:81
Method__getitem__
(self, idx)
code/utils.py:259
Method__getitem__
(self, idx)
code/utils.py:317
Method__getitem__
(self, idx)
code/SubgraphDataset.py:101
Method__getitem__
(self, idx)
code/SubgraphDataset.py:205
Method__init__
(self, data, args)
code/dataset_benchmark.py:27
Method__init__
(self, root, name, transform=None, pre_transform=None, random_seed=717, train_size=0.7, val_s
code/dataset_benchmark.py:111
Method__init__
(self, root, name, transform=None, pre_transform=None)
code/dataset_benchmark.py:270
Method__init__
(self, name, args, root='dataset')
code/dataset_benchmark.py:328
Method__init__
(self, root, name, args, transform=None, pre_transform=None)
code/dataset_benchmark.py:464
Method__init__
(self, features, label_emb, label)
code/utils.py:251
Method__init__
(self, input_dim: int, target_dim: int)
code/utils.py:263
Method__init__
(self, original_dataset)
code/utils.py:300
Method__init__
(self, args)
code/st_model.py:143
Method__init__
(self, data, num_hops=1, max_nodes=100, dataset_name="Arxiv", transform=None, pre_transform=None)
code/SubgraphDataset.py:148
Method__init__
(self, in_dim, emb_dim, out_dim, num_layer, act='relu')
code/model.py:15
Method__init__
(self, in_dim, emb_dim, num_layer, kernel='gcn', drop_ratio=0, act='relu', norm=None)
code/model.py:104
Method__init__
(self, in_dim, emb_dim, num_layer, kernel='gcn', drop_ratio=0, act='relu', norm=None)
code/model.py:156
Method__init__
(self, in_dim, emb_dim, num_layer, kernel='gcn', drop_ratio=0, act='relu', norm=None, concat=
code/model.py:208
Method__init__
(self, encoder)
code/model.py:280
Method__len__
(self)
code/utils.py:256
Method__len__
(self)
code/utils.py:314
Method__repr__
(self)
code/dataset_benchmark.py:200
Functionbuild_args
(task=None)
code/utils.py:38
Functioncreate_k_shot_mask
(labels, args)
code/utils.py:436
Functioncreate_logger
(args, task)
code/utils.py:128
Functioncycle_index
(num, shift)
code/utils.py:390
Methoddata
(self)
code/dataset_benchmark.py:41
Functiondo_CL
(X, Y, labels, args)
code/utils.py:407
Methoddownload
(self)
code/dataset_benchmark.py:142
Methoddownload
(self)
code/dataset_benchmark.py:233
Methoddownload
(self)
code/dataset_benchmark.py:297
Methodembed
(self, x, edge_index, edge_weigt=None, batch=None)
code/model.py:275
Methodfeature_embedding
(self)
code/dataset_benchmark.py:52
Functionformat_time
Takes a time in seconds and returns a string hh:mm:ss
code/main_TextBP_benchmark.py:36
Methodforward
(self, x: torch.Tensor)
code/utils.py:273
Methodforward
(self, x)
code/utils.py:290
Methodforward
(self, data)
code/st_model.py:33
Methodforward
(self, data,args)
code/st_model.py:201
Methodforward
(self, x)
code/model.py:39
Methodforward
(self, x, edge_index, edge_weight=None, batch=None)
code/model.py:82
Methodforward
(self, x, edge_index, edge_weigt=None, batch=None)
code/model.py:113
Methodforward
(self, x, edge_index, edge_weigt=None, batch=None)
code/model.py:172
Methodforward
(self, input)
code/model.py:306
Methodget_labels
(self, root)
code/dataset_benchmark.py:397
Functionget_logger
(output_dir)
code/main_TextBP_benchmark.py:45
Methodget_loss
(self, x, x_aug, batch_size)
code/model.py:121
Methodget_loss
(self, pos_h, neg_h, summary)
code/model.py:189
Methodget_loss
(self, x, edge_index, edge_weigt=None, batch=None)
code/model.py:268
Methodget_splits
(self)
code/dataset_benchmark.py:405
Functiongraphcl_augmentation
(features, edge_index)
code/utils.py:223
Methodlabel_embedding
(self)
code/dataset_benchmark.py:44
Methodlen
(self)
code/SubgraphDataset.py:95
Methodlen
(self)
code/SubgraphDataset.py:185
Functionnormalize_adjacency_matrix
(data,args)
code/utils.py:450
Methodprocess
(self)
code/dataset_benchmark.py:147
Methodprocess
(self)
code/dataset_benchmark.py:236
Methodprocess
(self)
code/dataset_benchmark.py:302
Methodprocess
(self)
code/dataset_benchmark.py:563
Methodprocessed_dir
(self)
code/dataset_benchmark.py:130
Methodprocessed_dir
(self)
code/dataset_benchmark.py:221
Methodprocessed_dir
(self)
code/dataset_benchmark.py:285
Methodprocessed_dir
(self)
code/dataset_benchmark.py:551
Methodprocessed_file_names
(self)
code/dataset_benchmark.py:139
Methodprocessed_file_names
(self)
code/dataset_benchmark.py:230
Methodprocessed_file_names
(self)
code/dataset_benchmark.py:294
Methodprocessed_file_names
(self)
code/dataset_benchmark.py:560
next →1–100 of 109, ranked by callers