python
import pylearn2
from pylearn2.models import mlp
from pylearn2.training_algorithms import sgd
from pylearn2.termination_criteria import EpochCounter
from pylearn2.datasets import cifar10
dataset = cifar10.CIFAR10(which_set='train')
model = mlp.MLP(
layers=[
mlp.Sigmoid(layer_name='h1', dim=100),
mlp.Sigmoid(layer_name='h2', dim=100),
mlp.Softmax(layer_name='y', n_classes=10)
],
nvis=dataset.X.shape[1]
)
algorithm = sgd.SGD(
learning_rate=0.01,
batch_size=100,
termination_criterion=EpochCounter(max_epochs=10)
)
trainer = pylearn2.train.Train(dataset=dataset, model=model, algorithm=algorithm)
trainer.main_loop()
