def write_stats()

in utils.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def write_stats(writer, niter, full_yp, labels):
    # compute accuracy across good and bad
    y = labels.flatten()
    acc_good = np.mean((full_yp[y < 0.5] < 0))
    acc_bad = np.mean((full_yp[y > 0.5] > 0))
    acc = (acc_good + acc_bad) / 2

    print(niter, acc, acc_good, acc_bad)

    writer.add_scalar("acc/yp", acc, niter)
    writer.add_scalar("acc/good" , acc_good, niter)
    writer.add_scalar("acc/bad" , acc_bad, niter)