python
import visvis as vv
dataset = vv.datasets.load_cifar10()
model = vv.nn.Sequential([
vv.nn.Conv2d(3, 32, 3, activation='relu'),
vv.nn.MaxPool2d(2),
vv.nn.Flatten(),
vv.nn.Linear(32*14*14, 128, activation='relu'),
vv.nn.Dropout(0.5),
vv.nn.Linear(128, 10, activation='softmax')
])
model.fit(dataset.train, epochs=10, batch_size=32)
accuracy = model.evaluate(dataset.test)
test_image = dataset.test[0][0]
prediction = model.predict(test_image)