4fca544205a55010e8959bb760aece5be80dc3a5,deeppavlov/models/evolution/check_binary_mask.py,,get_graph_and_plot,#Any#Any#Any#Any#,97

Before Change

    pos = {}

    for i in range(total_nodes):
        for j in range(total_nodes):
            if binary_mask[i, j] == 1:
                dg.add_edge(i, j)
        pos[i] = 2. * np.array(number_to_type_layer(i, n_types))[::-1]

    plt.figure(figsize=(12, 12))
    nx.draw(dg, pos, node_color="b", node_size=7000, alpha=0.3)

After Change

    plt.figure(figsize=(12, 12))
    values = [val_map.get(node, 0.25) for node in nodes]

    nx.draw(dg, pos, cmap=plt.get_cmap("jet"), node_color=values, node_size=7000, alpha=0.3)

    nx.draw_networkx_labels(dg, pos, nodes, font_size=18)

Instances