Skip to content

Commit

Permalink
Initial commit
Browse files Browse the repository at this point in the history
  • Loading branch information
joshua-robinson committed Mar 1, 2023
1 parent 2e569ab commit 8c63945
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions experiment_saver.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
from collections import defaultdict

import pandas as pd


class ExperimentSaver(defaultdict):

def __init__(self, save_fname):
super().__init__(list)
self.save_fname = save_fname

def save(self):
print("Saving to", self.save_fname)
pd.DataFrame(self).to_pickle(self.save_fname)

0 comments on commit 8c63945

Please sign in to comment.