Added creation of json files for each evaluation

dsdtools/evaluate.py

@@ -2,6 +2,8 @@
 import numpy as np
 import mir_eval
 import pandas as pd
+import json
+from os import path as op
 
 
 class DF_writer(object):
@@ -122,6 +124,26 @@ def evaluate_track(
             except ValueError:
                 pass
 
+            if estimates_dir and rows:
+                try:
+                    # save the dataframe corresponding to this estimate as json
+                    temp_frame = pd.DataFrame()
+                    for row in rows:
+                        temp_frame = temp_frame.append(row, ignore_index=True)
+                    json_string = json.dumps(json.loads(temp_frame.to_json()),
+                                             sort_keys=True,
+                                             indent=4,
+                                             separators=(',',':'))
+                    track_dir = op.join(
+                        estimates_dir,
+                        track.subset,
+                        track.filename
+                    )
+                    with open(track_dir + '/evaluation.json', 'w') as f:
+                        f.write(json_string)
+                except ValueError, IOError:
+                    pass
+
         return rows
 
     def evaluate(self, estimates, references):