def dump_eval(self, e, precision=2): assert isinstance(e, Evaluation) path = os.path.join( self.config.eval_dir, "{}-{}.json".format(e.data_type, str(e.global_step).zfill(6))) with open(path, 'w') as fh: json.dump(short_floats(e.dict, precision), fh)
def dump_eval(self, e, precision=2, path=None): assert isinstance(e, Evaluation) if self.config.dump_pickle: path = path or os.path.join(self.config.eval_dir, "{}-{}.pklz".format(e.data_type, str(e.global_step).zfill(6))) with gzip.open(path, 'wb', compresslevel=3) as fh: pickle.dump(e.dict, fh) else: path = path or os.path.join(self.config.eval_dir, "{}-{}.json".format(e.data_type, str(e.global_step).zfill(6))) with open(path, 'w') as fh: json.dump(short_floats(e.dict, precision), fh)