:param graph: :param embedding_size: :param negative_ratio: :param order: 'first','second','all'
(self, graph, embedding_size=8, negative_ratio=5, order='second', )
| 71 | |
| 72 | class LINE: |
| 73 | def __init__(self, graph, embedding_size=8, negative_ratio=5, order='second', ): |
| 74 | """ |
| 75 | |
| 76 | :param graph: |
| 77 | :param embedding_size: |
| 78 | :param negative_ratio: |
| 79 | :param order: 'first','second','all' |
| 80 | """ |
| 81 | if order not in ['first', 'second', 'all']: |
| 82 | raise ValueError('mode must be fisrt,second,or all') |
| 83 | |
| 84 | self.graph = graph |
| 85 | self.idx2node, self.node2idx = preprocess_nxgraph(graph) |
| 86 | self.use_alias = True |
| 87 | |
| 88 | self.rep_size = embedding_size |
| 89 | self.order = order |
| 90 | |
| 91 | self._embeddings = {} |
| 92 | self.negative_ratio = negative_ratio |
| 93 | self.order = order |
| 94 | |
| 95 | self.node_size = graph.number_of_nodes() |
| 96 | self.edge_size = graph.number_of_edges() |
| 97 | self.samples_per_epoch = self.edge_size * (1 + negative_ratio) |
| 98 | |
| 99 | self._gen_sampling_table() |
| 100 | self.reset_model() |
| 101 | |
| 102 | def reset_training_config(self, batch_size, times): |
| 103 | self.batch_size = batch_size |
nothing calls this directly
no test coverage detected