Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HazyResearch/hgcn
/ functions
Functions
178 in github.com/HazyResearch/hgcn
⨍
Functions
178
◇
Types & classes
37
↓ 10 callers
Method
proj
Projects point p on the manifold.
manifolds/base.py:23
↓ 9 callers
Method
_lambda_x
(self, x, c)
manifolds/poincare.py:33
↓ 6 callers
Method
__init__
(self, c, args)
models/encoders.py:37
↓ 6 callers
Method
forward
(self, x)
layers/layers.py:70
↓ 5 callers
Method
encode
(self, x, adj)
models/encoders.py:71
↓ 5 callers
Method
expmap0
Exponential map of u at the origin.
manifolds/base.py:43
↓ 5 callers
Method
forward
(self, x, adj)
layers/hyp_layers.py:134
↓ 5 callers
Method
proj_tan0
Projects u on the tangent space of the origin.
manifolds/base.py:31
↓ 4 callers
Method
__init__
(self, manifold, c, in_features, dropout, use_att, local_agg)
layers/hyp_layers.py:122
↓ 4 callers
Function
artanh
(x)
utils/math_utils.py:26
↓ 4 callers
Method
compute_metrics
(self, embeddings, data, split)
models/base_models.py:77
↓ 4 callers
Function
format_metrics
Format metric in metric dict for logging.
utils/train_utils.py:9
↓ 4 callers
Function
get_dim_act
Helper function to get dimension and activation at every layer. :param args: :return:
layers/layers.py:11
↓ 4 callers
Method
minkowski_dot
(self, x, y, keepdim=True)
manifolds/hyperboloid.py:25
↓ 3 callers
Method
__init__
(self, c)
models/decoders.py:15
↓ 3 callers
Method
logmap0
Logarithmic map of point p at the origin.
manifolds/base.py:47
↓ 3 callers
Method
mobius_add
(self, x, y, c, dim=-1)
manifolds/poincare.py:85
↓ 3 callers
Method
proj_tan
(self, u, x, c)
manifolds/hyperboloid.py:54
↓ 3 callers
Function
tanh
(x, clamp=15)
utils/math_utils.py:14
↓ 2 callers
Method
__init__
(self, in_features, out_features, dropout, act, use_bias)
layers/layers.py:64
↓ 2 callers
Method
__init__
(self, in_features, dropout)
layers/att_layers.py:9
↓ 2 callers
Method
__init__
(self, args)
models/base_models.py:59
↓ 2 callers
Method
_gyration
(self, u, v, w, c, dim: int = -1)
manifolds/poincare.py:108
↓ 2 callers
Function
arcosh
(x)
utils/math_utils.py:18
↓ 2 callers
Function
copy_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 callers
Function
cosh
(x, clamp=15)
utils/math_utils.py:6
↓ 2 callers
Method
decode
(self, h, idx)
models/base_models.py:103
↓ 2 callers
Method
expmap
Exponential map of u at point p.
manifolds/base.py:35
↓ 2 callers
Function
load_citation_data
(dataset_str, use_feats, data_path, split_seed=None)
utils/data_utils.py:173
↓ 2 callers
Function
load_data_airport
(dataset_str, data_path, return_label=False)
utils/data_utils.py:244
↓ 2 callers
Function
load_data_lp
(dataset, use_feats, data_path)
utils/data_utils.py:133
↓ 2 callers
Function
load_synthetic_data
(dataset_str, use_feats, data_path)
utils/data_utils.py:211
↓ 2 callers
Method
logmap
(self, x, y, c)
manifolds/hyperboloid.py:80
↓ 2 callers
Method
logmap0
(self, x, c)
manifolds/hyperboloid.py:103
↓ 2 callers
Method
minkowski_norm
(self, u, keepdim=True)
manifolds/hyperboloid.py:31
↓ 2 callers
Function
normalize
Row-normalize sparse matrix.
utils/data_utils.py:49
↓ 2 callers
Method
proj
(self, x, c)
manifolds/hyperboloid.py:43
↓ 2 callers
Function
sinh
(x, clamp=15)
utils/math_utils.py:10
↓ 2 callers
Method
sqdist
(self, x, y, c)
manifolds/hyperboloid.py:35
↓ 2 callers
Method
step
Performs a single optimization step. Arguments --------- closure : callable (optional) A closure that reevaluates
optimizers/radam.py:75
↓ 1 callers
Function
OrNone
(default)
utils/train_utils.py:53
↓ 1 callers
Function
acc_f1
(output, labels, average='binary')
utils/eval_utils.py:3
↓ 1 callers
Function
add_flags_from_config
Adds a flag (and default value) to an ArgumentParser for each parameter in a config
utils/train_utils.py:48
↓ 1 callers
Function
augment
(adj, features, normalize_feats=True)
utils/data_utils.py:70
↓ 1 callers
Method
backward
(ctx, grad_output)
utils/math_utils.py:39
↓ 1 callers
Function
bin_feat
(feat, bins)
utils/data_utils.py:125
↓ 1 callers
Method
decode
(self, x, adj)
models/decoders.py:19
↓ 1 callers
Method
decode
(self, h, adj, idx)
models/base_models.py:73
↓ 1 callers
Method
egrad2rgrad
Converts Euclidean Gradient to Riemannian Gradients.
manifolds/base.py:19
↓ 1 callers
Method
expmap
(self, u, x, c)
manifolds/hyperboloid.py:70
↓ 1 callers
Method
expmap0
(self, u, c)
manifolds/hyperboloid.py:90
↓ 1 callers
Function
get_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 callers
Method
has_improved
(self, m1, m2)
models/base_models.py:88
↓ 1 callers
Function
hyperbolicity_sample
(G, num_samples=50000)
utils/hyperbolicity.py:13
↓ 1 callers
Method
init_metric_dict
(self)
models/base_models.py:85
↓ 1 callers
Method
init_weights
Initializes random weigths on the manifold.
manifolds/base.py:59
↓ 1 callers
Method
inner
Inner product for tangent vectors at point x.
manifolds/base.py:63
↓ 1 callers
Function
load_data
(args, datapath)
utils/data_utils.py:12
↓ 1 callers
Function
load_data_nc
(dataset, use_feats, data_path, split_seed)
utils/data_utils.py:149
↓ 1 callers
Method
logmap
Logarithmic map of point p1 at point p2.
manifolds/base.py:39
↓ 1 callers
Function
mask_edges
(adj, val_prop, test_prop, seed)
utils/data_utils.py:82
↓ 1 callers
Method
mobius_add
Adds points x and y.
manifolds/base.py:51
↓ 1 callers
Method
mobius_matvec
Performs hyperboic martrix-vector multiplication.
manifolds/base.py:55
↓ 1 callers
Method
normalize
(self, p)
manifolds/euclidean.py:15
↓ 1 callers
Function
parse_index_file
(filename)
utils/data_utils.py:204
↓ 1 callers
Function
process
(adj, features, normalize_adj, normalize_feats)
utils/data_utils.py:37
↓ 1 callers
Method
proj_tan
Projects u on the tangent space of p.
manifolds/base.py:27
↓ 1 callers
Method
ptransp
Parallel transport of u from x to y.
manifolds/base.py:67
↓ 1 callers
Method
ptransp0
(self, x, u, c)
manifolds/hyperboloid.py:133
↓ 1 callers
Method
reset_parameters
(self)
layers/hyp_layers.py:95
↓ 1 callers
Function
sparse_mx_to_torch_sparse_tensor
Convert a scipy sparse matrix to a torch sparse tensor.
utils/data_utils.py:59
↓ 1 callers
Function
split_data
(labels, val_prop, test_prop, seed)
utils/data_utils.py:105
↓ 1 callers
Method
sqdist
Squared distance between pairs of points.
manifolds/base.py:15
↓ 1 callers
Method
stabilize_group
(self, group)
optimizers/radam.py:14
↓ 1 callers
Method
stabilize_group
(self, group)
optimizers/radam.py:163
↓ 1 callers
Function
train
(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