Skip to content

Instantly share code, notes, and snippets.

@jkarnows
Created September 3, 2017 19:54
Show Gist options
  • Save jkarnows/60fb809769a292c33b2fba1858d1fd64 to your computer and use it in GitHub Desktop.
Save jkarnows/60fb809769a292c33b2fba1858d1fd64 to your computer and use it in GitHub Desktop.
Plotting training loss
import os
loss_files = sort(os.listdir('cv/'))
lfs = []
for lf in loss_files:
lf = map(float,lf[13:-3].split('_'))
lfs.append(lf)
lfs = array(sorted(lfs, key = lambda x: x[0]))
plot(lfs[:,0],lfs[:,1])
title('Training Lossn')
ylabel('Loss')
xlabel('Epoch')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment