Plot metrics logged by PL CSVLogger

plot_metrics

df = pd.read_csv(csv)
df.drop(columns=['step', 'epoch'], inplace=True)
df.fillna(method='ffill', inplace=True)
df.fillna(method='bfill', inplace=True)
df.drop_duplicates(inplace=True)
df.reset_index(inplace=True, drop=True)
df = df.iloc[::2,:].reset_index(drop=True)

fig, ax = plt.subplots(1,2,figsize=(15,5))
df[['train_loss', 'val_loss']].plot(ax=ax[0])
ax[0].set_xlabel('Epoch')
ax[0].set_ylabel('Loss')

df[['train_accuracy', 'val_accuracy']].plot(ax=ax[1])
ax[1].set_xlabel('Epoch')
ax[1].set_ylabel('Accuracy')