def split_test_val()

in svhn_based_exp/utils.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def split_test_val(testloader):
    testset = testloader.dataset
    batch_size = testloader.batch_size
    img_inds = np.arange(len(testset))
    np.random.seed(0)
    np.random.shuffle(img_inds)
    test_inds = img_inds[:int(0.5 * len(img_inds))]
    val_inds = img_inds[int(0.5 * len(img_inds)):]
    testloader = data_utils.DataLoader(testset, num_workers=2, batch_size=batch_size, sampler=data_utils.SubsetRandomSampler(test_inds))
    valloader = data_utils.DataLoader(testset, num_workers=2, batch_size=batch_size, sampler=data_utils.SubsetRandomSampler(val_inds))
    np.random.seed(int(time.time()))
    return testloader, valloader