5508d6730a37f9aff49549901ff8682898e14b5d,brats/evaluate.py,,main,#,27

Before Change

    plt.ylabel("Loss")
    plt.xlabel("Epoch")
    plt.xlim((0, len(training_df.index)))
    plt.legend(loc="upper right")
    plt.savefig("loss_graph.png")


if __name__ == "__main__":

After Change

    plt.savefig("validation_scores_boxplot.png")
    plt.close()

    if os.path.exists("./training.log"):
        training_df = pd.read_csv("./training.log").set_index("epoch")

        plt.plot(training_df["loss"].values, label="training loss")
        plt.plot(training_df["val_loss"].values, label="validation loss")
        plt.ylabel("Loss")
        plt.xlabel("Epoch")
        plt.xlim((0, len(training_df.index)))
        plt.legend(loc="upper right")
        plt.savefig("loss_graph.png")


if __name__ == "__main__":
    main()

Instances