()
| 11 | |
| 12 | |
| 13 | def test_DeepWalk(): |
| 14 | graph = nx.read_edgelist( |
| 15 | str(TEST_GRAPH_PATH), |
| 16 | create_using=nx.DiGraph(), |
| 17 | nodetype=None, |
| 18 | data=[("weight", int)], |
| 19 | ) |
| 20 | |
| 21 | model = DeepWalk(graph, walk_length=3, num_walks=2, workers=1) |
| 22 | model.train(embed_size=8, window_size=2, iter=1, workers=1) |
| 23 | embeddings = model.get_embeddings() |
| 24 | assert len(embeddings) == graph.number_of_nodes() |
| 25 | assert all(len(vector) == 8 for vector in embeddings.values()) |
| 26 | |
| 27 | |
| 28 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected