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()