graphs = [nx.newman_watts_strogatz_graph(50, 5, 0.3) for _ in range(1000)] model = Graph2Vec() model.fit(graphs) model.get_embedding() #----------------------------------- # BoostNE example #----------------------------------- g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = BoostNE() model.fit(g) model.get_embedding() #----------------------------------- # Diff2Vec example #----------------------------------- g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = Diff2Vec() model.fit(g) model.get_embedding() #----------------------------------- # NMF ADMM example
"""BoostNE illustrative example.""" import networkx as nx from karateclub.node_embedding.neighbourhood import BoostNE g = nx.newman_watts_strogatz_graph(100, 20, 0.05) model = BoostNE() model.fit(g)