[plot loss acc] #python #pytorch #keras

plot_loss_acc.py

import os
import seaborn as sns
import matplotlib.pyplot as plt

def plot_history(history, save_path:str=None, figsize = (20, 9)):
    f, axes = plt.subplots(1, 2, figsize=figsize)
    sns.lineplot(range(1, history.epochs+1), history.history['acc'], label='Train Accuracy', ax=axes[0])
    sns.lineplot(range(1, history.epochs+1), history.history['val_acc'], label='Val Accuracy', ax=axes[0])
    sns.lineplot(range(1, history.epochs+1), history.history['loss'], label='Train Loss', ax=axes[1])
    sns.lineplot(range(1, history.epochs+1), history.history['val_loss'], label='Val Loss', ax=axes[1])
    plt.tight_layout()
    if save_path is not None:
        f.savefig(os.path.join(save_path+"history_output.png"))
    else:
        plt.show()
plot_history(history)