cogdl.models.nn.dgl_gcc
¶
Module Contents¶
Functions¶
|
|
|
one epoch training for moco |
|
|
|
|
|
-
cogdl.models.nn.dgl_gcc.
_add_undirected_graph_positional_embedding
(g, hidden_size, retry=10)[source]¶
-
cogdl.models.nn.dgl_gcc.
_rwr_trace_to_dgl_graph
(g, seed, trace, positional_embedding_size, entire_graph=False)[source]¶
-
class
cogdl.models.nn.dgl_gcc.
NodeClassificationDataset
(data, rw_hops=64, subgraph_size=64, restart_prob=0.8, positional_embedding_size=32, step_dist=[1.0, 0.0, 0.0])[source]¶ Bases:
object