cogdl.models.emb.line
¶
Module Contents¶
Classes¶
The LINE model from the `”Line: Large-scale information network embedding” |
-
class
cogdl.models.emb.line.
LINE
(dimension, walk_length, walk_num, negative, batch_size, alpha, order)[source]¶ Bases:
cogdl.models.BaseModel
The LINE model from the “Line: Large-scale information network embedding” paper.
- Args:
hidden_size (int) : The dimension of node representation. walk_length (int) : The walk length. walk_num (int) : The number of walks to sample for each node. negative (int) : The number of nagative samples for each edge. batch_size (int) : The batch size of training in LINE. alpha (float) : The initial learning rate of SGD. order (int) : 1 represents perserving 1-st order proximity, 2 represents 2-nd, while 3 means both of them (each of them having dimension/2 node representation).