In [1]:
from keras.models import Sequential
model = Sequential()
In [2]:
from keras.layers.core import Dense, Activation
model.add(Dense(output_dim=64, input_dim=100, init="glorot_uniform"))
model.add(Activation("relu"))
model.add(Dense(output_dim=10, init="glorot_uniform"))
model.add(Activation("softmax"))
In [3]:
model.compile(loss='categorical_crossentropy', optimizer='sgd')
In [5]:
from keras.optimizers import SGD
model.compile(loss='categorical_crossentropy', optimizer=SGD(lr=0.01, momentum=0.9, nesterov=True))
model.fit(X_train, Y_train, nb_epoch=5, batch_size=32)
In [6]:
model.fit(X_train, Y_train, nb_epoch=5, batch_size=32)
In [ ]: