def test()

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

• 25 lines of code
• 3 McCabe index (conditional complexity)

def test(fields):
    N = args.N

    print("N =", N)
    logDir = "logs/grasp"
    os.makedirs(logDir, exist_ok=True)

    print(fields)
    accs = []
    # for i in range(K):
    for i in range(K):
        st = time.time()

        print("Fold {} of {}".format(i, K), end=" ")

        learner = Learning(K, i, fields=fields)
        lossTrainList, accTestList = learner.train(10)

        fn = "{}/field{}_N{}_i{}.h5".format(logDir, fields, N, i)

        dd.io.save(fn, {"lossTrainList": lossTrainList, "accTestList": accTestList})

        acc = learner.evaluation()
        print(acc, end=" ")

        print("time {:.3f}s".format(time.time() - st))
        accs.append(acc)

        modelDir = "models/grasp"
        os.makedirs(modelDir, exist_ok=True)
        fnModel = "{}/field{}_N{}_i{}.pth".format(modelDir, fields, N, i)

        learner.model.save(fnModel)

        # learner.model.load(fnModel)
        # acc = learner.evaluation()
        del learner

    print(accs)
    print("{:.2f}% ± {:.2f}%".format(np.mean(accs) * 100, np.std(accs) * 100))