<|file_name|>output2csv.py<|end_file_name|><｜fim▁begin｜>import os, re, csv

# regular expressions for capturing the interesting quantities
noise_pattern = 'noise: \[(.+)\]'
res_pattern = '^([0-9.]+$)'

search_dir = "output"
results_file = '../results.csv'

os.chdir( search_dir )
files = filter( os.path.isfile, os.listdir( '.' ))
#files = [ os.path.join( search_dir, f ) for f in files ] # add path to each file
files.sort( key=lambda x: os.path.getmtime( x ))

results = []

for file in files:
	f = open( file )
	contents = f.read()
	
	# noise
	matches = re.search( noise_pattern, contents, re.DOTALL )
	try:
		noise = matches.group( 1 )
		noise = noise.strip()
		noise = noise.split()
		
	except AttributeError:<｜fim▁hole｜>		print "noise error 1: %s" % ( contents )
		continue		
	
	# rmse
	matches = re.search( res_pattern, contents, re.M )
	try:
		res = matches.group( 1 )
	except AttributeError:
		print "matches error 2: %s" % ( contents )
		continue
		
	results.append( [ res ] + noise )
	
writer = csv.writer( open( results_file, 'wb' ))
for result in results:
	writer.writerow( result )<｜fim▁end｜>