import numpy as np import matplotlib.pyplot as plt from keras.datasets import mnist from keras.layers import Activation, Dense from keras.models import Sequential, load_model from keras import optimizers from keras.utils.np_utils import to_categorical (X_train, y_train), (X_test, y_test) = mnist.load_data() X_train = X_train.reshape(X_train.shape[0], 784)[:6000] X_test = X_test.reshape(X_test.shape[0], 784)[:1000] y_train = to_categorical(y_train)[:6000] y_test = to_categorical(y_test)[:1000] model = Sequential() model.add(Dense(256, input_dim=784)) model.add(Activation("sigmoid")) model.add(Dense(128)) model.add(Activation("sigmoid")) model.add(Dense(10)) model.add(Activation("softmax")) model.compile(optimizer="sgd", loss="categorical_crossentropy", metrics=["accuracy"]) model.fit(X_train, y_train, verbose=True) score = model.evaluate(X_test, y_test, verbose=True) print("evaluate loss: {0[0]}\nevaluate acc: {0[1]}".format(score))