select-iteration.py

# python3 -W ignore select-iteration.py PATH METRIC

import os
import sys
from event_file_loader import EventFileLoader

def extract(path):
    loader = EventFileLoader(path)
    x = []
    for e in loader.Load():
        try:
            if e.summary.value[0].tag == sys.argv[2]:
                x.append(e.summary.value[0].simple_value)
        except (AttributeError, IndexError):
            pass
    return x

b = extract('%s/b/%s' % (sys.argv[1], os.listdir('%s/b' % sys.argv[1])[0]))
c = extract('%s/c/%s' % (sys.argv[1], os.listdir('%s/c' % sys.argv[1])[0]))

# TODO tensorboard bug?
min_len = min(len(b), len(c))
b = b[:min_len]
c = c[:min_len]
inf = float('Inf')
predicate = lambda pair: pair[0] != inf and pair[1] != inf
try:
    b, c = map(list, zip(*filter(predicate, zip(b, c))))
except ValueError:
    pass

if b:
    x = max(b)
    print('%.3f %.3f' % (x, c[b.index(x)]))
else:
    print('0.0 0.0')