def get_train_valid_data(config): dataset = config.dataset num_bits = config.num_bits valid_frac = config.valid_frac return get_data(dataset, num_bits, train=True, valid_frac=valid_frac, pad=config.pad)
def get_test_data(dataset, num_bits): return get_data(dataset, num_bits, train=False)
def get_train_valid_data(dataset, num_bits, valid_frac): return get_data(dataset, num_bits, train=True, valid_frac=valid_frac)