import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import os
import numpy as np
import cPickle as pickle
import time
import sys

filename = sys.argv[1]
file = open(filename)

last_chunk = -1
training_errors = []
validation_errors = []
training_idcs = []
validation_idcs=[]

for line in file:
	if 'Chunk' in line :
		last_chunk = int(line.split()[1].split('/')[0])
	if 'Validation loss' in line:
		validation_errors.append(float(line.split(':')[1].rsplit()[0]))
		validation_idcs.append(last_chunk)
	if 'Mean train loss' in line:
		training_errors.append(float(line.split(':')[1].rsplit()[0]))
		training_idcs.append(last_chunk)


print 'training errors'
print training_errors
print training_idcs
print 'validation errors'
print validation_errors
print validation_idcs

print 'min training error', np.amin(np.array(training_errors)), 'at', np.argmin(np.array(training_errors))
print 'min validation error', np.amin(np.array(validation_errors)), 'at', np.argmin(np.array(validation_errors))

plt.plot(training_errors, label='training errors')
plt.plot(validation_errors, label='validation errors')
plt.legend(loc="upper right")
plt.title(sys.argv[1])
plt.xlabel('Epoch')
#plt.ylim(0, 0.7)
plt.ylabel('Error')	
plt.savefig(sys.argv[2])