Skip to content

Instantly share code, notes, and snippets.

@jkarnows

jkarnows/loss_plot.py

Created September 3, 2017 19:54
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jkarnows/60fb809769a292c33b2fba1858d1fd64 to your computer and use it in GitHub Desktop.
Save jkarnows/60fb809769a292c33b2fba1858d1fd64 to your computer and use it in GitHub Desktop.
Download ZIP
Plotting training loss
Raw
loss_plot.py
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