(self, graph, hidden_size=[32, 16], alpha=1e-6, beta=5., nu1=1e-5, nu2=1e-4, )
| 80 | |
| 81 | class SDNE(object): |
| 82 | def __init__(self, graph, hidden_size=[32, 16], alpha=1e-6, beta=5., nu1=1e-5, nu2=1e-4, ): |
| 83 | |
| 84 | self.graph = graph |
| 85 | # self.g.remove_edges_from(self.g.selfloop_edges()) |
| 86 | self.idx2node, self.node2idx = preprocess_nxgraph(self.graph) |
| 87 | |
| 88 | self.node_size = self.graph.number_of_nodes() |
| 89 | self.hidden_size = hidden_size |
| 90 | self.alpha = alpha |
| 91 | self.beta = beta |
| 92 | self.nu1 = nu1 |
| 93 | self.nu2 = nu2 |
| 94 | |
| 95 | self.A, self.L = _create_A_L(self.graph, self.node2idx) # Adj Matrix,L Matrix |
| 96 | self.reset_model() |
| 97 | self._embeddings = {} |
| 98 | |
| 99 | def reset_model(self, opt='adam'): |
| 100 |
nothing calls this directly
no test coverage detected