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

    pos = {}
    val_map = {}
    sources, sinks = find_sources_and_sinks(dg)

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

Instances