Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/beabevi/ESAN
/ functions
Functions
105 in github.com/beabevi/ESAN
⨍
Functions
105
◇
Types & classes
26
↓ 7 callers
Function
policy2transform
(policy: str, num_hops, process_subgraphs=lambda x: x, pbar=None)
data.py:525
↓ 5 callers
Method
__init__
(self, num_tasks, num_layer=5, in_dim=300, emb_dim=300, gnn_type='gin', num_random_features=0
models.py:23
↓ 5 callers
Function
preprocess
(dataset, transform)
data.py:163
↓ 4 callers
Method
__init__
emb_dim (int): node embedding dimensionality
conv.py:16
↓ 4 callers
Method
separate_data
(self, seed, fold_idx)
data.py:78
↓ 3 callers
Method
__init__
(self, fraction)
data.py:95
↓ 3 callers
Function
eval
(model, device, loader, evaluator, voting_times=1)
main.py:41
↓ 2 callers
Method
eval
(self, input_dict)
utils.py:201
↓ 2 callers
Function
subgraph_pool
(h_node, batched_data, pool)
models.py:9
↓ 2 callers
Function
to_undirected
r"""Converts the graph given by :attr:`edge_index` to an undirected graph such that :math:`(j,i) \in \mathcal{E}` for every edge :math:`(i,j) \in
data.py:123
↓ 1 callers
Function
S2V_to_PyG
(data)
data.py:334
↓ 1 callers
Method
acc
(self, input_dict)
utils.py:154
↓ 1 callers
Function
get_data
(args, fold_idx)
utils.py:18
↓ 1 callers
Function
get_model
(args, in_dim, out_dim, device)
utils.py:102
↓ 1 callers
Function
load_data
dataset: name of dataset test_proportion: ratio of test train split seed: random seed for random splitting of dataset
data.py:344
↓ 1 callers
Method
mae
(self, input_dict)
utils.py:169
↓ 1 callers
Function
main
()
data.py:540
↓ 1 callers
Function
main
()
main.py:164
↓ 1 callers
Method
process_CSL
(self)
csl_data.py:90
↓ 1 callers
Function
reset_wandb_env
()
main.py:69
↓ 1 callers
Function
run
(args, device, fold_idx, sweep_run_name, sweep_id, results_queue)
main.py:80
↓ 1 callers
Method
to_subgraphs
(self, data)
data.py:213
↓ 1 callers
Function
train
(model, device, loader, optimizer, criterion, epoch, fold_idx)
main.py:19
↓ 1 callers
Function
unbatch_subgraphs
(data)
data.py:164
Method
__call__
(self, data)
data.py:98
Method
__call__
(self, data)
data.py:197
Method
__inc__
(self, key, value)
data.py:115
Method
__init__
(self, task_type)
utils.py:151
Method
__init__
(self, num_tasks)
utils.py:183
Method
__init__
(self, num_tasks, num_layer=5, in_dim=300, emb_dim=300, gnn_type='gin', num_random_features=0
models.py:66
Method
__init__
(self, subgraph_gnn, channels, num_tasks, invariant)
models.py:94
Method
__init__
(self, num_layers, in_dim, emb_dim, num_tasks, feature_encoder, GNNConv)
models.py:149
Method
__init__
(self, encoder)
models.py:210
Method
__init__
(self, policy, emb_dim)
models.py:220
Method
__init__
(self, root: str, name: str, split: str = "train", transform: Optional[Callable] = None,
csl_data.py:22
Method
__init__
(self, root: str, name: str, transform=None, pre_transform=None,
data.py:38
Method
__init__
(self, process_subgraphs=lambda x: x, pbar=None)
data.py:193
Method
__init__
(self, num_hops, add_node_idx=False, process_subgraphs=lambda x: x, pbar=None)
data.py:279
Method
__init__
g: a networkx graph label: an integer graph label node_tags: a list of integer node tags node_feature
data.py:315
Method
__init__
( self, root, name, transform=None, pre_transform=
data.py:440
Method
__init__
(self, in_dim, emb_dim)
conv.py:43
Method
__init__
(self, in_dim, emb_dim)
conv.py:66
Method
__init__
(self, in_dim, emb_dim)
conv.py:82
Method
__init__
emb_dim (int): node embedding dimensionality num_layer (int): number of GNN message passing layers
conv.py:118
Method
__init__
(self, root, name=None, transform=None, pre_transform=None, pre_filter=None)
gnn_rni_data.py:15
Method
__repr__
(self)
csl_data.py:136
Method
download
(self)
csl_data.py:76
Method
download
(self)
data.py:60
Method
download
(self)
data.py:483
Method
download
(self)
gnn_rni_data.py:33
Method
eval
(self, input_dict)
utils.py:177
Method
eval
(self, input_dict)
utils.py:186
Method
eval_metric
(self)
csl_data.py:116
Method
eval_metric
(self)
data.py:53
Method
eval_metric
(self)
data.py:468
Method
eval_metric
(self)
gnn_rni_data.py:62
Method
filter
(self, record)
data.py:30
Method
forward
(self, batched_data)
models.py:59
Method
forward
(self, batched_data)
models.py:86
Method
forward
(self, batched_data)
models.py:120
Method
forward
(self, batched_data)
models.py:179
Method
forward
(self, x)
models.py:215
Method
forward
(self, x)
models.py:226
Method
forward
(self, x, edge_index, edge_attr)
conv.py:29
Method
forward
(self, x, edge_index, edge_attr)
conv.py:52
Method
forward
(self, x, edge_index, edge_attr)
conv.py:76
Method
forward
(self, x, edge_index, edge_attr)
conv.py:89
Method
forward
(self, batched_data)
conv.py:169
Method
message
(self, x_j, edge_attr)
conv.py:35
Method
message
(self, x_j, edge_attr)
conv.py:58
Method
message
(self, x_j, edge_attr, norm)
conv.py:104
Method
num_node_labels
(self)
data.py:68
Method
num_tasks
(self)
csl_data.py:112
Method
num_tasks
(self)
data.py:49
Method
num_tasks
(self)
data.py:464
Method
num_tasks
(self)
gnn_rni_data.py:58
Method
process
(self)
csl_data.py:83
Method
process
(self)
data.py:63
Method
process
(self)
data.py:495
Method
process
(self)
gnn_rni_data.py:36
Method
processed_dir
(self)
csl_data.py:55
Method
processed_dir
(self)
data.py:459
Method
processed_dir
(self)
gnn_rni_data.py:25
Method
processed_file_names
(self)
csl_data.py:70
Method
processed_file_names
(self)
data.py:480
Method
processed_file_names
(self)
gnn_rni_data.py:30
Method
raw_dir
(self)
csl_data.py:51
Method
raw_dir
(self)
data.py:454
Method
raw_file_names
(self)
csl_data.py:59
Method
raw_file_names
(self)
data.py:476
Method
raw_file_names
(self)
gnn_rni_data.py:21
Method
separate_data
(self, seed, fold_idx)
csl_data.py:123
Method
separate_data
(self, seed, fold_idx)
data.py:509
Method
separate_data
(self, seed, fold_idx)
gnn_rni_data.py:69
Method
task_type
(self)
csl_data.py:120
Method
task_type
(self)
data.py:57
Method
task_type
(self)
data.py:472
Method
task_type
(self)
gnn_rni_data.py:66
Method
to_subgraphs
(self, data)
data.py:218
Method
to_subgraphs
(self, data)
data.py:259
next →
1–100 of 105, ranked by callers