[keras] misc keras things #keras

plot_keras.py

# see https://github.com/stared/livelossplot for live plots, or tensorboard
# but for simple stuff the below is good enough

def plot_history(history, log=True):
    """takes in a keras history object and plots train and val loss and accuracy"""
    
    # dict which stores train & val accuracy and losses over epochs
    hist = history.history
    
    fig, (ax, ax2) = plt.subplots(1,2, figsize=(13,6))
    fig.suptitle("Train/Val History of Loss and Accuracy", fontsize=15)
    
    # setup subplots
    ax.set_title("Loss against Epochs")
    ax.set_xlabel('number of epoches', fontsize = 13)
    if log: 
        ax.set_yscale('log')
        ax.set_ylabel('loss (log)', fontsize = 13)
    else:
        ax.set_ylabel('loss', fontsize = 13)

    ax2.set_title("Accuracy against Epochs")
    ax2.set_xlabel('number of epoches', fontsize = 13)
    ax2.set_ylabel('Accuracy', fontsize = 13);
    
    # plots train loss and accuracy
    x = history.epoch
    ax.plot(x, hist["loss"], label=f"train_loss")
    ax2.plot(x, hist["accuracy"], label=f"train_accuracy")
    
    # plots val data - in try as we might have no validation data
    try: 
        ax2.plot(x, hist["val_accuracy"], label=f"val_accuracy")
        ax.plot(x, hist["val_loss"], label="val_loss")
    except:
        pass
    
    ax.legend(); ax2.legend()
    plt.show()