示例#1
0
文件: images.py 项目: johnpjust/nsf
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)
示例#2
0
文件: images.py 项目: xukai92/nsf
def get_test_data(dataset, num_bits):
    return get_data(dataset, num_bits, train=False)
示例#3
0
文件: images.py 项目: xukai92/nsf
def get_train_valid_data(dataset, num_bits, valid_frac):
    return get_data(dataset, num_bits, train=True, valid_frac=valid_frac)