MCPcopy Create free account

hub / github.com/cornell-zhang/Polynormer / functions

Functions45 in github.com/cornell-zhang/Polynormer

⨍ Functions 45◇ Types & classes 6
↓ 13 callersMethodreset_parameters
(self)
large_graph_exp/lg_model.py:31
↓ 11 callersMethodreset_parameters
(self)
model.py:36
↓ 6 callersMethodprint_statistics
(self, run=None, mode='max_acc')
logger.py:14
↓ 6 callersFunctionsave_model
(args, model, optimizer, run)
logger.py:70
↓ 3 callersMethodadd_result
(self, run, result)
logger.py:9
↓ 3 callersFunctionload_dataset
Loader for NCDataset Returns NCDataset
dataset.py:65
↓ 3 callersFunctionload_model
(args, model, optimizer, run)
logger.py:78
↓ 3 callersFunctionsave_result
(args, results)
logger.py:86
↓ 2 callersFunctionevaluate
(model, dataset, split_idx, eval_func, criterion, args, result=None)
eval.py:5
↓ 2 callersFunctionload_fixed_splits
(data_dir, dataset, name)
data_utils.py:8
↓ 2 callersFunctionparse_method
(args, n, c, d, device)
large_graph_exp/lg_parse.py:3
↓ 2 callersFunctionparser_add_main_args
(parser)
large_graph_exp/lg_parse.py:11
↓ 1 callersMethod__init__
(self, hidden_channels, heads, num_layers, beta, dropout, qk_shared=True)
model.py:6
↓ 1 callersMethod__init__
(self, hidden_channels, heads, num_layers, beta, dropout, qk_shared=True)
large_graph_exp/lg_model.py:6
↓ 1 callersFunctionevaluate_cpu
(model, dataset, split_idx, eval_func, criterion, args, device, result=None)
eval.py:34
↓ 1 callersFunctionfix_seed
(seed=42)
main.py:16
↓ 1 callersFunctionfix_seed
(seed=42)
large_graph_exp/main-arxiv.py:18
↓ 1 callersFunctionfix_seed
(seed=42)
large_graph_exp/main-batch.py:19
↓ 1 callersMethodget_idx_split
train_prop: The proportion of dataset for train split. Between 0 and 1. valid_prop: The proportion of dataset for validation split. B
dataset.py:46
↓ 1 callersFunctionload_amazon_dataset
(data_dir, name)
dataset.py:133
↓ 1 callersFunctionload_coauthor_dataset
(data_dir, name)
dataset.py:158
↓ 1 callersFunctionload_hetero_dataset
(data_dir, name)
dataset.py:105
↓ 1 callersFunctionload_ogb_dataset
(data_dir, name)
dataset.py:183
↓ 1 callersFunctionload_pokec_mat
requires pokec.mat
dataset.py:199
↓ 1 callersFunctionload_wikics_dataset
(data_dir)
dataset.py:87
↓ 1 callersFunctionparse_method
(args, n, c, d, device)
parse.py:3
↓ 1 callersFunctionparser_add_main_args
(parser)
parse.py:10
Method__getitem__
(self, idx)
dataset.py:54
Method__init__
based off of ogb NodePropPredDataset https://github.com/snap-stanford/ogb/blob/master/ogb/nodeproppred/dataset.py Gives torch
dataset.py:17
Method__init__
(self, runs, info=None)
logger.py:5
Method__init__
(self, in_channels, hidden_channels, out_channels, local_layers=3, global_layers=2, in_dropout=0.1
model.py:88
Method__init__
(self, in_channels, hidden_channels, out_channels, local_layers=3, global_layers=2, in_dropout=0.1
large_graph_exp/lg_model.py:76
Method__len__
(self)
dataset.py:58
Method__repr__
(self)
dataset.py:61
Functioneval_acc
(y_true, y_pred)
data_utils.py:59
Functioneval_f1
(y_true, y_pred)
data_utils.py:48
Functioneval_rocauc
adapted from ogb https://github.com/snap-stanford/ogb/blob/master/ogb/nodeproppred/evaluate.py
data_utils.py:71
Methodforward
(self, x)
model.py:54
Methodforward
(self, x, edge_index)
model.py:148
Methodforward
(self, x)
large_graph_exp/lg_model.py:45
Methodforward
(self, x, edge_index)
large_graph_exp/lg_model.py:161
Functionogb_idx_to_tensor
()
dataset.py:190
Methodoutput
(self,out_path,info)
logger.py:64
Methodreset_parameters
(self)
model.py:126
Methodreset_parameters
(self)
large_graph_exp/lg_model.py:139