cogdl.models.nn.mvgrl
¶
Module Contents¶
Classes¶
Functions¶
|
Row-normalize feature matrix and convert to tuple representation |
|
Symmetrically normalize adjacency matrix. |
|
Convert a scipy sparse matrix to a torch sparse tensor. |
|
-
cogdl.models.nn.mvgrl.
preprocess_features
(features)[source]¶ Row-normalize feature matrix and convert to tuple representation
-
cogdl.models.nn.mvgrl.
sparse_mx_to_torch_sparse_tensor
(sparse_mx)[source]¶ Convert a scipy sparse matrix to a torch sparse tensor.
-
class
cogdl.models.nn.mvgrl.
MVGRL
(nfeat, nhid, nclass, max_epochs)[source]¶ Bases:
cogdl.models.BaseModel