Python NetLSD.get_embedding Examples

Example #1

def test_netlsd():
    """
    Test the NetLSD embedding.
    """
    graphs = [nx.newman_watts_strogatz_graph(50, 5, 0.3) for _ in range(100)]

    model = NetLSD()

    model.fit(graphs)
    embedding = model.get_embedding()

    assert embedding.shape[0] == len(graphs)
    assert embedding.shape[1] == model.scale_steps
    assert type(embedding) == np.ndarray

    graphs = [nx.newman_watts_strogatz_graph(500, 5, 0.3) for _ in range(100)]

    model = NetLSD()

    model.fit(graphs)
    embedding = model.get_embedding()

    assert embedding.shape[0] == len(graphs)
    assert embedding.shape[1] == model.scale_steps
    assert type(embedding) == np.ndarray

Example #2

File: netlsd_example.py Project: zyloong/karateclub

"""NetLSD illustrative example."""

import networkx as nx
from karateclub.graph_embedding import NetLSD

graphs = [nx.newman_watts_strogatz_graph(50, 5, 0.3) for _ in range(100)]

model = NetLSD()

model.fit(graphs)
model.get_embedding()