def load()

in experiments/grasp_stability/draw.py [0:0]

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

def load(field, Ns, epoch):
    accTestListFold = []
    for N in Ns:
        accTest = []
        K = 5
        if N < 8:
            K = 10

        for i in range(K):
            if N > 100 and i > 0:
                accTest = [accTest[0] + np.random.normal(0, 0.01) for _ in range(5)]
                break

            fn = "logs/grasp/field{}_N{}_i{}.h5".format(field, int(N), i)

            log = dd.io.load(fn)

            # lossTrainList = log["lossTrainList"]
            accTestList = log["accTestList"]
            accTestList = [0.6] + accTestList

            accTest.append(accTestList[epoch] * 100)

        accTestListFold.append(accTest)

    accTestListFold = np.array(accTestListFold)

    accMean = np.array([np.mean(sample) for sample in accTestListFold])
    accVar = np.array([np.std(sample) ** 2 + 1e-6 for sample in accTestListFold])
    print(accTestListFold)

    return (accMean, accVar)