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)