MCPcopy Create free account

hub / github.com/HazyResearch/hgcn / functions

Functions178 in github.com/HazyResearch/hgcn

↓ 10 callersMethodproj
Projects point p on the manifold.
manifolds/base.py:23
↓ 9 callersMethod_lambda_x
(self, x, c)
manifolds/poincare.py:33
↓ 6 callersMethod__init__
(self, c, args)
models/encoders.py:37
↓ 6 callersMethodforward
(self, x)
layers/layers.py:70
↓ 5 callersMethodencode
(self, x, adj)
models/encoders.py:71
↓ 5 callersMethodexpmap0
Exponential map of u at the origin.
manifolds/base.py:43
↓ 5 callersMethodforward
(self, x, adj)
layers/hyp_layers.py:134
↓ 5 callersMethodproj_tan0
Projects u on the tangent space of the origin.
manifolds/base.py:31
↓ 4 callersMethod__init__
(self, manifold, c, in_features, dropout, use_att, local_agg)
layers/hyp_layers.py:122
↓ 4 callersFunctionartanh
(x)
utils/math_utils.py:26
↓ 4 callersMethodcompute_metrics
(self, embeddings, data, split)
models/base_models.py:77
↓ 4 callersFunctionformat_metrics
Format metric in metric dict for logging.
utils/train_utils.py:9
↓ 4 callersFunctionget_dim_act
Helper function to get dimension and activation at every layer. :param args: :return:
layers/layers.py:11
↓ 4 callersMethodminkowski_dot
(self, x, y, keepdim=True)
manifolds/hyperboloid.py:25
↓ 3 callersMethod__init__
(self, c)
models/decoders.py:15
↓ 3 callersMethodlogmap0
Logarithmic map of point p at the origin.
manifolds/base.py:47
↓ 3 callersMethodmobius_add
(self, x, y, c, dim=-1)
manifolds/poincare.py:85
↓ 3 callersMethodproj_tan
(self, u, x, c)
manifolds/hyperboloid.py:54
↓ 3 callersFunctiontanh
(x, clamp=15)
utils/math_utils.py:14
↓ 2 callersMethod__init__
(self, in_features, out_features, dropout, act, use_bias)
layers/layers.py:64
↓ 2 callersMethod__init__
(self, in_features, dropout)
layers/att_layers.py:9
↓ 2 callersMethod__init__
(self, args)
models/base_models.py:59
↓ 2 callersMethod_gyration
(self, u, v, w, c, dim: int = -1)
manifolds/poincare.py:108
↓ 2 callersFunctionarcosh
(x)
utils/math_utils.py:18
↓ 2 callersFunctioncopy_or_set_
A workaround to respect strides of :code:`dest` when copying :code:`source` (https://github.com/geoopt/geoopt/issues/70) Parameters -
optimizers/radam.py:24
↓ 2 callersFunctioncosh
(x, clamp=15)
utils/math_utils.py:6
↓ 2 callersMethoddecode
(self, h, idx)
models/base_models.py:103
↓ 2 callersMethodexpmap
Exponential map of u at point p.
manifolds/base.py:35
↓ 2 callersFunctionload_citation_data
(dataset_str, use_feats, data_path, split_seed=None)
utils/data_utils.py:173
↓ 2 callersFunctionload_data_airport
(dataset_str, data_path, return_label=False)
utils/data_utils.py:244
↓ 2 callersFunctionload_data_lp
(dataset, use_feats, data_path)
utils/data_utils.py:133
↓ 2 callersFunctionload_synthetic_data
(dataset_str, use_feats, data_path)
utils/data_utils.py:211
↓ 2 callersMethodlogmap
(self, x, y, c)
manifolds/hyperboloid.py:80
↓ 2 callersMethodlogmap0
(self, x, c)
manifolds/hyperboloid.py:103
↓ 2 callersMethodminkowski_norm
(self, u, keepdim=True)
manifolds/hyperboloid.py:31
↓ 2 callersFunctionnormalize
Row-normalize sparse matrix.
utils/data_utils.py:49
↓ 2 callersMethodproj
(self, x, c)
manifolds/hyperboloid.py:43
↓ 2 callersFunctionsinh
(x, clamp=15)
utils/math_utils.py:10
↓ 2 callersMethodsqdist
(self, x, y, c)
manifolds/hyperboloid.py:35
↓ 2 callersMethodstep
Performs a single optimization step. Arguments --------- closure : callable (optional) A closure that reevaluates
optimizers/radam.py:75
↓ 1 callersFunctionOrNone
(default)
utils/train_utils.py:53
↓ 1 callersFunctionacc_f1
(output, labels, average='binary')
utils/eval_utils.py:3
↓ 1 callersFunctionadd_flags_from_config
Adds a flag (and default value) to an ArgumentParser for each parameter in a config
utils/train_utils.py:48
↓ 1 callersFunctionaugment
(adj, features, normalize_feats=True)
utils/data_utils.py:70
↓ 1 callersMethodbackward
(ctx, grad_output)
utils/math_utils.py:39
↓ 1 callersFunctionbin_feat
(feat, bins)
utils/data_utils.py:125
↓ 1 callersMethoddecode
(self, x, adj)
models/decoders.py:19
↓ 1 callersMethoddecode
(self, h, adj, idx)
models/base_models.py:73
↓ 1 callersMethodegrad2rgrad
Converts Euclidean Gradient to Riemannian Gradients.
manifolds/base.py:19
↓ 1 callersMethodexpmap
(self, u, x, c)
manifolds/hyperboloid.py:70
↓ 1 callersMethodexpmap0
(self, u, c)
manifolds/hyperboloid.py:90
↓ 1 callersFunctionget_dir_name
Gets a directory to save the model. If the directory already exists, then append a new integer to the end of it. This method is useful so tha
utils/train_utils.py:15
↓ 1 callersMethodhas_improved
(self, m1, m2)
models/base_models.py:88
↓ 1 callersFunctionhyperbolicity_sample
(G, num_samples=50000)
utils/hyperbolicity.py:13
↓ 1 callersMethodinit_metric_dict
(self)
models/base_models.py:85
↓ 1 callersMethodinit_weights
Initializes random weigths on the manifold.
manifolds/base.py:59
↓ 1 callersMethodinner
Inner product for tangent vectors at point x.
manifolds/base.py:63
↓ 1 callersFunctionload_data
(args, datapath)
utils/data_utils.py:12
↓ 1 callersFunctionload_data_nc
(dataset, use_feats, data_path, split_seed)
utils/data_utils.py:149
↓ 1 callersMethodlogmap
Logarithmic map of point p1 at point p2.
manifolds/base.py:39
↓ 1 callersFunctionmask_edges
(adj, val_prop, test_prop, seed)
utils/data_utils.py:82
↓ 1 callersMethodmobius_add
Adds points x and y.
manifolds/base.py:51
↓ 1 callersMethodmobius_matvec
Performs hyperboic martrix-vector multiplication.
manifolds/base.py:55
↓ 1 callersMethodnormalize
(self, p)
manifolds/euclidean.py:15
↓ 1 callersFunctionparse_index_file
(filename)
utils/data_utils.py:204
↓ 1 callersFunctionprocess
(adj, features, normalize_adj, normalize_feats)
utils/data_utils.py:37
↓ 1 callersMethodproj_tan
Projects u on the tangent space of p.
manifolds/base.py:27
↓ 1 callersMethodptransp
Parallel transport of u from x to y.
manifolds/base.py:67
↓ 1 callersMethodptransp0
(self, x, u, c)
manifolds/hyperboloid.py:133
↓ 1 callersMethodreset_parameters
(self)
layers/hyp_layers.py:95
↓ 1 callersFunctionsparse_mx_to_torch_sparse_tensor
Convert a scipy sparse matrix to a torch sparse tensor.
utils/data_utils.py:59
↓ 1 callersFunctionsplit_data
(labels, val_prop, test_prop, seed)
utils/data_utils.py:105
↓ 1 callersMethodsqdist
Squared distance between pairs of points.
manifolds/base.py:15
↓ 1 callersMethodstabilize_group
(self, group)
optimizers/radam.py:14
↓ 1 callersMethodstabilize_group
(self, group)
optimizers/radam.py:163
↓ 1 callersFunctiontrain
(args)
train.py:20
Method__init__
(self, in_features, out_features, dropout, act, use_bias)
layers/layers.py:34
Method__init__
(self, r, t)
layers/layers.py:80
Method__init__
(self, in_features, out_features, dropout, alpha, activation)
layers/att_layers.py:65
Method__init__
Sparse version of GAT.
layers/att_layers.py:121
Method__init__
(self, manifold, in_features, out_features, c, dropout, act, use_bias)
layers/hyp_layers.py:47
Method__init__
(self, manifold, in_features, out_features, c_in, c_out, dropout, act, use_bias, use_att, local_agg)
layers/hyp_layers.py:63
Method__init__
(self, manifold, in_features, out_features, c, dropout, use_bias)
layers/hyp_layers.py:83
Method__init__
(self, manifold, c_in, c_out, act)
layers/hyp_layers.py:164
Method__init__
(self)
manifolds/euclidean.py:11
Method__init__
(self)
manifolds/hyperboloid.py:18
Method__init__
(self)
manifolds/base.py:11
Method__init__
(self, data, requires_grad, manifold, c)
manifolds/base.py:83
Method__init__
(self, )
manifolds/poincare.py:19
Method__init__
(self, *args, stabilize=None, **kwargs)
optimizers/radam.py:10
Method__init__
(self, c)
models/encoders.py:20
Method__init__
(self, c, args)
models/encoders.py:55
Method__init__
(self, c, args)
models/encoders.py:80
Method__init__
(self, c, args)
models/encoders.py:98
Method__init__
(self, c, args)
models/encoders.py:129
Method__init__
(self, c, args)
models/encoders.py:152
Method__init__
(self, c, args)
models/decoders.py:33
Method__init__
(self, c, args)
models/decoders.py:45
Method__init__
(self, c, args)
models/decoders.py:56
Method__init__
(self, args)
models/base_models.py:22
next →1–100 of 178, ranked by callers