示例#1
0
def train():
    image_paths = [
        os.path.join('emoji', name) for name in os.listdir('emoji')
        if name.endswith('.png')
    ]
    images = np.array([load_image(path, 64) for path in image_paths])
    train_batcher = batch_generator(images, images, size=20, random=True)
    test_batcher = batch_generator(images, images, size=40, random=True)
    n.training_loop(train_batcher, test_batcher)
示例#2
0
def train():
    test_in, test_out = load_dataset('t10k', noise=True)
    train_in, train_out = load_dataset('train', noise=True)
    train_batcher = batch_generator(train_in, train_in, size=64, random=True)
    test_batcher = batch_generator(test_in, test_in, size=128, random=True)
    n.training_loop(train_batcher, test_batcher, evaluation_interval=10)