Files
graph_recognition_w_attn/plot_results.py