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)


上一篇:
下一篇:
切换中文