Example #1
0
def test_hybrid_node_edge_sampler():
    """
    Testing the edge retention rate.
    """
    sampler = HybridNodeEdgeSampler()

    graph = nx.watts_strogatz_graph(200, 10, 0)

    new_graph = sampler.sample(graph)

    assert sampler.number_of_edges == new_graph.number_of_edges()
Example #2
0
def test_hybrid_node_edge_sampler():
    """
    Testing the edge retention rate.
    """
    sampler = HybridNodeEdgeSampler()

    graph = nx.watts_strogatz_graph(200, 10, 0)

    new_graph = sampler.sample(graph)

    assert sampler.number_of_edges == new_graph.number_of_edges()
    assert type(new_graph) == NXGraph

    sampler = HybridNodeEdgeSampler(number_of_edges=25)

    graph = nx.watts_strogatz_graph(100, 10, 0)

    new_graph = sampler.sample(graph)

    assert sampler.number_of_edges == new_graph.number_of_edges()
    assert type(new_graph) == NXGraph

    sampler = HybridNodeEdgeSampler()

    graph = nk.generators.WattsStrogatzGenerator(200, 10, 0.0).generate()

    new_graph = sampler.sample(graph)

    assert sampler.number_of_edges == new_graph.numberOfEdges()
    assert type(new_graph) == NKGraph

    sampler = HybridNodeEdgeSampler(number_of_edges=25)

    graph = nk.generators.WattsStrogatzGenerator(200, 10, 0.0).generate()

    new_graph = sampler.sample(graph)

    assert sampler.number_of_edges == new_graph.numberOfEdges()
    assert type(new_graph) == NKGraph
Example #3
0
new_graph = sampler.sample(graph)

#----------------------------------
# Random Node Edge Sampler Example
#----------------------------------

sampler = RandomNodeEdgeSampler()

new_graph = sampler.sample(graph)

#----------------------------------
# Hybrid Node Edge Sampler Example
#----------------------------------

sampler = HybridNodeEdgeSampler()

new_graph = sampler.sample(graph)

#--------------------------------------
# Random Node Neighbor Sampler Example
#--------------------------------------

sampler = RandomNodeNeighborSampler()

new_graph = sampler.sample(graph)

#------------------------------------------
# Random Walk With Restart Sampler Example
#------------------------------------------
"""Hybrid node edge sampler example."""

import networkx as nx
from littleballoffur.edge_sampling import HybridNodeEdgeSampler

graph = nx.watts_strogatz_graph(1000, 10, 0)

sampler = HybridNodeEdgeSampler()

new_graph = sampler.sample(graph)