compare_histories.py

def plot_histories(*args):
  '''
  Plots histories
  '''
  epochs1 = np.arange(1, len(pd.DataFrame(args[0].history)) + 1)
  fig, ax = plt.subplots(len(args), 2, figsize=(16, 7 * len(args)))
  fig.suptitle('Comparing Model Histories')
  for i in range(len(args)):
    loss1 = pd.DataFrame(args[i].history["loss"])
    ax[i, 0].set_title(f"Model {i + 1} Loss")
    ax[i, 0].plot(np.arange(1, len(pd.DataFrame(args[i].history)) + 1), args[i].history["loss"], np.arange(1, len(pd.DataFrame(args[i].history)) + 1), args[i].history["val_loss"])
    ax[i, 0].set_xlabel('Epochs')
    ax[i, 0].legend(["Train Loss", "Val Loss"])
    ax[i, 1].set_title(f"Model {i + 1} Accuracy")
    ax[i, 1].plot(np.arange(1, len(pd.DataFrame(args[i].history)) + 1), args[i].history["accuracy"], np.arange(1, len(pd.DataFrame(args[i].history)) + 1), args[i].history["val_accuracy"])
    ax[i, 1].set_xlabel('Epochs')
    ax[i, 1].legend(["Train Accuracy", "Val Accuracy"]);