save_to_csv.py

import csv
import numpy as np
import pickle
import pprint


def save_results(results, filename):
    with open(filename[:-4] + ".pkl", "wb") as f:
        pickle.dump(results, f)
        print(results)
        pp = pprint.PrettyPrinter()
        pp.pprint(results)

    percentages = list(results[0].keys())
    measures = list(list(results[0].values())[0].keys())

    f = open(filename, 'wt')
    writer = csv.writer(f)

    for m in measures:
        writer.writerow((m,))
        header = ['Percentage %d' % x for x in percentages]
        writer.writerow(header)

        for r in results:
            row = []
            for p in percentages:
                if p in r:
                    row.append(r[p][m])
            writer.writerow(tuple(row))

    f.close()

    mean_f1 = np.asarray([r[50]['f1'] for r in results]).mean()

    f = open(filename[:-4] + "_%.3f" % mean_f1, 'w')
    f.close()

    print('Mean F1 at 50%%: %.3f' % mean_f1)