Add visualisation.

print('[a] Loading ' + sys.argv[0] + '.\n')

import config

import json

import matplotlib.pyplot as plt

import networkx as nx # for visualisation

import numpy as np

import os # for reading files

        graph.node[node]['dist'] = np.mean(graph.node[node]['dist'], axis=0)

    draw_propagation(graph,root_hub_list)

    return graph

    return mapping_dict

def draw_propagation(graph: nx.Graph, root_hub_list: list) -> None:

    pass

    pos = nx.spring_layout(G)

    mapping = {node:graph.node[node]['sense'] for node in graph.nodes}

    nx.draw_networkx_edges(G, pos, alpha=0.4)

    nx.draw_networkx_nodes(G,pos,

                           nodelist=list(mapping.keys()),

                           node_color=list(mapping.values()),

                           node_size=80,

                           cmap=plt.cm.hsv)

    labels={root:root for root in root_hub_list}

    nx.draw_networkx_labels(G,pos,labels,font_size=14)

    plt.axis('off')

    plt.savefig('../figures/'+''.join([r[:2] for r in root_hub_list])+'.png')

    plt.clf()

##############################

#     MST Disambiguation     #