MCPcopy Create free account

hub / github.com/EdisonLeeeee/MaskGAE / functions

Functions47 in github.com/EdisonLeeeee/MaskGAE

⨍ Functions 47◇ Types & classes 7
↓ 5 callersMethodreset_parameters
(self)
maskgae/model.py:313
↓ 4 callersMethod__init__
( self, encoder, edge_decoder, degree_decoder=None, mask=None,
maskgae/model.py:279
↓ 4 callersMethodbatch_predict
(self, z, edges, batch_size=2 ** 16)
maskgae/model.py:384
↓ 4 callersMethodtest_step
(self, data, pos_edge_index, neg_edge_index, batch_size=2**16)
maskgae/model.py:393
↓ 3 callersFunctioncreat_activation_layer
(activation)
maskgae/model.py:61
↓ 3 callersFunctionset_seed
(seed: int)
maskgae/utils.py:12
↓ 3 callersFunctiontab_printer
Function to print the logs in a nice tabular format. Note ---- Package `Texttable` is required. Run `pip install Texttable` if was no
maskgae/utils.py:66
↓ 3 callersMethodtest_step_ogb
(self, data, evaluator, pos_edge_index, neg_edge_index, batch_size=2**16)
maskgae/model.py:409
↓ 3 callersMethodtrain_step
(self, data, optimizer, alpha=0.002, batch_size=2 ** 16, grad_norm=1.0 )
maskgae/model.py:323
↓ 2 callersMethodcreate_input_feat
(self, x)
maskgae/model.py:123
↓ 2 callersFunctionget_dataset
(root: str, name: str, transform=None)
maskgae/utils.py:23
↓ 2 callersFunctiontest
(loader)
train_nodeclas.py:62
↓ 2 callersFunctionto_sparse_tensor
(edge_index, num_nodes)
maskgae/model.py:25
↓ 1 callersMethod__init__
(self, p: float = 0.7, walks_per_node: int = 1, walk_length: int = 3,
maskgae/mask.py:68
↓ 1 callersFunctioncreat_gnn_layer
(name, first_channels, second_channels, heads)
maskgae/model.py:31
↓ 1 callersFunctioncreate_input_layer
(num_nodes, num_node_feats, use_node_feats=True, node_emb=None)
maskgae/model.py:47
↓ 1 callersMethodget_embedding
(self, x, edge_index, mode="cat")
maskgae/model.py:148
↓ 1 callersFunctionmask_edge
(edge_index: Tensor, p: float=0.7)
maskgae/mask.py:57
↓ 1 callersFunctionmask_path
(edge_index: Tensor, p: float = 0.3, walks_per_node: int = 1, walk_length: int = 3, num_nodes: O
maskgae/mask.py:16
↓ 1 callersFunctiontrain_linkpred
(model, splits, args, device="cpu")
train_nodeclas.py:18
↓ 1 callersFunctiontrain_linkpred
(model, splits, args, device="cpu")
train_linkpred_ogb.py:16
↓ 1 callersFunctiontrain_linkpred
(model, splits, args, device="cpu")
train_linkpred.py:14
↓ 1 callersFunctiontrain_nodeclas
(model, data, args, device='cpu')
train_nodeclas.py:60
Method__init__
(self, p: float=0.7, undirected: bool=True)
maskgae/mask.py:98
Method__init__
( self, in_channels, hidden_channels, out_channels, num_layers=2,
maskgae/model.py:73
Method__init__
(self, *args, **kwargs)
maskgae/model.py:179
Method__init__
( self, in_channels, hidden_channels, out_channels=1, num_layers=2, dropout=0.5, activation='r
maskgae/model.py:198
Method__init__
( self, in_channels, hidden_channels, out_channels=1, num_layers=2, dropout=0.5, activation='r
maskgae/model.py:240
Functionauc_loss
(pos_out, neg_out)
maskgae/loss.py:4
Functionce_loss
(pos_out, neg_out)
maskgae/loss.py:13
Methodextra_repr
(self)
maskgae/mask.py:92
Methodextra_repr
(self)
maskgae/mask.py:109
Methodforward
(self, edge_index)
maskgae/mask.py:82
Methodforward
(self, edge_index)
maskgae/mask.py:103
Methodforward
(self, x, edge_index)
maskgae/model.py:132
Methodforward
(self, z, edge, sigmoid=True)
maskgae/model.py:185
Methodforward
(self, z, edge, sigmoid=True, reduction=False)
maskgae/model.py:218
Methodforward
(self, x)
maskgae/model.py:260
Methodforward
(self, x, edge_index)
maskgae/model.py:320
Functionhinge_auc_loss
(pos_out, neg_out)
maskgae/loss.py:7
Functioninfo_nce_loss
(pos_out, neg_out)
maskgae/loss.py:18
Functionlog_rank_loss
(pos_out, neg_out, num_neg=1)
maskgae/loss.py:10
Functionrandom_negative_sampler
(edge_index, num_nodes, num_neg_samples)
maskgae/model.py:273
Methodreset_parameters
(self)
maskgae/model.py:112
Methodreset_parameters
(self)
maskgae/model.py:182
Methodreset_parameters
(self)
maskgae/model.py:214
Methodreset_parameters
(self)
maskgae/model.py:256