def test_scd():
"""
Test the SCD procedure.
"""
graph = nx.newman_watts_strogatz_graph(50, 5, 0.3)
model = SCD()
model.fit(graph)
memberships = model.get_memberships()
indices = [k for k, v in memberships.items()].sort()
nodes = [node for node in graph.nodes()].sort()
assert graph.number_of_nodes() == len(memberships)
assert indices == nodes
assert type(memberships) == dict
graph = nx.newman_watts_strogatz_graph(150, 5, 0.3)
model = SCD()
model.fit(graph)
memberships = model.get_memberships()
indices = [k for k, v in memberships.items()].sort()
nodes = [node for node in graph.nodes()].sort()
assert graph.number_of_nodes() == len(memberships)
assert indices == nodes
assert type(memberships) == dict
from karateclub.graph_embedding import Graph2Vec, FGSD, GL2Vec, SF from karateclub.node_embedding.attributed import BANE, TENE, TADW, FSCNMF from karateclub.node_embedding.structural import GraphWave, Role2Vec from karateclub.dataset import GraphReader, GraphSetReader #------------------------------------ # SCD example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 10, 0.2) model = SCD() model.fit(g) model.get_memberships() #------------------------------------ # Symm-NMF example #------------------------------------ g = nx.newman_watts_strogatz_graph(100, 10, 0.2) model = SymmNMF() model.fit(g) model.get_embedding() model.get_memberships()
