def test_bane():
"""
Testing the BANE node embedding.
"""
graph = nx.newman_watts_strogatz_graph(250, 10, 0.2)
features = np.random.uniform(0, 1, (250, 256))
model = BANE()
model.fit(graph, features)
embedding = model.get_embedding()
assert embedding.shape[0] == graph.number_of_nodes()
assert embedding.shape[1] == model.dimensions
assert type(embedding) == np.ndarray
graph = nx.newman_watts_strogatz_graph(150, 10, 0.2)
features = np.random.uniform(0, 1, (150, 256))
model = BANE(dimensions=8)
model.fit(graph, features)
embedding = model.get_embedding()
assert embedding.shape[0] == graph.number_of_nodes()
assert embedding.shape[1] == model.dimensions
assert type(embedding) == np.ndarray
quit() #------------------------------------ # Walklets example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = Walklets() model.fit(g) emb = model.get_embedding() print(emb.shape) #------------------------------------ # DeepWalk example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = DeepWalk() model.fit(g)
model.fit(g) membership = model.get_memberships() #------------------------------------ # Walklets example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = Walklets() model.fit(g) emb = model.get_embedding() #------------------------------------ # DeepWalk example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = DeepWalk() model.fit(g) #------------------------------------ # Splitter example #------------------------------------