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
model.fit(g, tp) model.get_embedding() #--------------- # BANE example #--------------- g = nx.newman_watts_strogatz_graph(100, 20, 0.05) x = np.random.uniform(0,1,(100,2000)) p = nx.newman_watts_strogatz_graph(100, 20, 0.05) x = nx.adjacency_matrix(p) model = BANE() model.fit(g, x) #----------------- # BigClam example #----------------- g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = BigClam() model.fit(g) membership = model.get_memberships()
"""BANE example.""" import numpy as np import networkx as nx from karateclub.node_embedding.attributed import BANE g = nx.newman_watts_strogatz_graph(200, 20, 0.05) x = np.random.uniform(0, 1, (200, 200)) model = BANE() model.fit(g, x)
from karateclub.node_embedding.structural import GraphWave from karateclub.node_embedding.attributed import BANE #----------------------------------- # BANE example #----------------------------------- g = nx.newman_watts_strogatz_graph(100, 20, 0.05) x = np.random.uniform(0,1,(100,2000)) p = nx.newman_watts_strogatz_graph(100, 20, 0.05) x = nx.adjacency_matrix(p) model = BANE() model.fit(g, x) quit() #------------------------------------ # BigClam example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = BigClam() model.fit(g)
from karateclub.node_embedding.neighbourhood import GraRep, DeepWalk, Walklets from karateclub.node_embedding.structural import GraphWave from karateclub.node_embedding.attributed import BANE #----------------------------------- # BANE example #----------------------------------- g = nx.newman_watts_strogatz_graph(100, 20, 0.05) x = np.random.uniform(0, 1, (100, 2000)) p = nx.newman_watts_strogatz_graph(100, 20, 0.05) x = nx.adjacency_matrix(p) model = BANE() model.fit(g, x) #------------------------------------ # BigClam example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = BigClam() model.fit(g) membership = model.get_memberships()