def max_across_seed()

in pyrela/parse_log.py [0:0]

• 15 lines of code
• 7 McCabe index (conditional complexity)

def max_across_seed(logs):
    new_logs = {}
    for k, v in logs.items():
        s = k.rsplit("_", 1)
        if len(s) == 2:
            name, seed = s
        elif len(s) == 1:
            name = "default"
            seed = s[0]
        if not seed.startswith("SEED"):
            print("no multiple seeds, omit averaging: ", name)
            name = k
        if name not in new_logs or np.mean(v[-10:]) > new_logs[name][0]:
            new_logs[name] = (np.mean(v[-10:]), k)

    return new_logs