net1 = NeuralNet(
layers=[ # three layers: one hidden layer
('input', layers.InputLayer),
('hidden', layers.DenseLayer),
('output', layers.DenseLayer),
],
# layer parameters:
input_shape=(None, 400), # 20x20 input pixels per batch
hidden_num_units=300, # number of units in hidden layer
output_nonlinearity=lasagne.nonlinearities.softmax, # output layer uses identity function
output_num_units=60, # 1 target values
# optimization method:
update=nesterov_momentum,
update_learning_rate=0.01,
update_momentum=0.9,
regression=False, # flag to indicate we're dealing with regression problem
max_epochs=100, # we want to train this many epochs
verbose=1,
)