Plotting the loss function using a log scale using Matplotlib

plot-loss.py

import matplotlib.pyplot as plt
from matplotlib import rcParams

rcParams['figure.figsize'] = (12, 10)
colors = plt.rcParams['axes.prop_cycle'].by_key()['color']

def plot_log_loss(history: History, title_label: str, n: int) -> ():
    # Use a log scale to show the wide range of values.
    plt.semilogy(history.epoch,  history.history['loss'],
               color=colors[n], label='Train '+title_label)
    plt.semilogy(history.epoch,  history.history['val_loss'],
          color=colors[n], label='Val '+title_label,
          linestyle="--")
    plt.xlabel('Epoch')
    plt.ylabel('Loss')

    plt.legend()

plot_log_loss(history, "Model Base", 1)