def expVis(X, res_pre, m_summ, node_labels=None, di_graph=None):
print('\tGraph Visualization:')
if node_labels:
node_colors = plot_util.get_node_color(node_labels)
else:
node_colors = None
plot_embedding2D(X, node_colors=node_colors,
di_graph=di_graph)
plt.savefig('%s_%s_vis.pdf' % (res_pre, m_summ), dpi=300,
format='pdf', bbox_inches='tight')
plt.figure()
def plot_single_step_v2(node_pos, graph_info, graph_info_next, purturbed_nodes,
dyn_changed_node):
node_colors = plot_util.get_node_color(graph_info_next[1])
node_num, embedding_dimension = node_pos.shape
pos = {}
for i in range(node_num):
pos[i] = node_pos[i, :]
# unchanged_nodes = list(set(range(node_num)) - set(purturbed_nodes) - set(dyn_changed_node))
unchanged_nodes = list(set(range(node_num)) - set(dyn_changed_node))
nodes_draw = nx.draw_networkx_nodes(
graph_info[0],
pos,
nodelist=unchanged_nodes,
node_color=[node_colors[p] for p in unchanged_nodes],
node_size=40,
with_labels=False)
nodes_draw.set_edgecolor('w')
# nodes_draw=nx.draw_networkx_nodes(graph_info[0],
# pos,
# nodelist=graph_info_next[4],
# node_color='r',
# node_size=50,
# with_labels=False)
# nodes_draw.set_edgecolor('w')
# nodes_draw=nx.draw_networkx_nodes(graph_info[0],
# pos,
# nodelist=graph_info_next[5],
# node_color='g',
# node_size=50,
# with_labels=False)
# nodes_draw.set_edgecolor('w')
# nodes_draw=nx.draw_networkx_nodes(graph_info[0],
# pos,
# nodelist=purturbed_nodes,
# node_color='r',
# node_size=80,
# with_labels=False)
# nodes_draw.set_edgecolor('k')
nodes_draw = nx.draw_networkx_nodes(graph_info[0],
pos,
nodelist=dyn_changed_node,
node_color='r',
node_size=80,
with_labels=False)
nodes_draw.set_edgecolor('k')
def plot_single_step(node_pos, graph, node_labels):
if node_labels:
node_colors = plot_util.get_node_color(node_labels)
else:
node_colors = None
node_num, embedding_dimension = node_pos.shape
pos = {}
for i in range(node_num):
pos[i] = node_pos[i, :]
# draw nodes
nx.draw_networkx_nodes(graph,
pos,
nodelist=range(node_num),
node_color=node_colors,
node_size=20,
with_labels=False)
# draw all edges
nx.draw_networkx_edges(graph, pos, width=0.1, arrows=False, alpha=0.8)
def plot_single_step(node_pos, graph_info, dyn_changed_node):
node_colors= plot_util.get_node_color(graph_info[1])
node_num, embedding_dimension = node_pos.shape
pos = {}
for i in range(node_num):
pos[i] = node_pos[i, :]
unchanged_nodes = list(set(range(node_num)) - set(dyn_changed_node))
nodes_draw=nx.draw_networkx_nodes(graph_info[0],
pos,
nodelist=unchanged_nodes,
node_color=[node_colors[p] for p in unchanged_nodes],
node_size=40,
with_labels=False)
nodes_draw.set_edgecolor('w')
nodes_draw=nx.draw_networkx_nodes(graph_info[0],
pos,
nodelist=dyn_changed_node,
node_color='r',
node_size=80,
with_labels=False)
print('Error reading file: {0}. Cannot load previous weights'.
format('decoder_weights_' + filesuffix + '.hdf5'))
exit()
if node_l is not None:
return decoder.predict(embed, batch_size=self._n_batch)[:,
node_l]
else:
return decoder.predict(embed, batch_size=self._n_batch)
if __name__ == '__main__':
# load Zachary's Karate graph
edge_f = 'data/karate.edgelist'
my_graph = SBM_graph.SBMGraph(1000, 2)
my_graph.sample_graph()
node_colors = plot_util.get_node_color(my_graph._node_community)
t1 = time()
embedding = SDNE(d=128,
beta=5,
alpha=0,
nu1=1e-6,
nu2=1e-6,
K=3,
n_units=[500, 300],
n_iter=20,
xeta=0.01,
n_batch=500,
modelfile=[
'./intermediate/enc_model.json',
'./intermediate/dec_model.json'
],