plt.savefig("images/confusion_matrix.png")