def _export_scalars(args): with _open_file(args.export_scalars) as f: out = csv.writer(f) out.writerow(["run", "path", "tag", "value", "step"]) for run in _list_runs_cb(args)(): for path, _digest, reader in tfevent.scalar_readers(run.dir): subpath = os.path.relpath(path, run.dir) for tag, value, step in reader: out.writerow([run.id, subpath, tag, value, step])
def _runs_scalars_detail(runs): from guild import tfevent data = [] cols = [ "run", "path", "tag", "val", "step", ] for run in runs: for path, _run_id, scalars in tfevent.scalar_readers(run.dir): rel_path = os.path.relpath(path, run.dir) for tag, val, step in scalars: data.append([run, rel_path, tag, val, step]) return pd.DataFrame(data, columns=cols)
def _iter_scalar_tags(dir): for _path, _digest, scalars in tfevent.scalar_readers(dir): for s in scalars: yield s[0]
def refresh(self, runs): for run in runs: for path, cur_digest, scalars in tfevent.scalar_readers(run.path): self._maybe_refresh_run_scalars(run, path, cur_digest, scalars)