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