from keras.datasets import mnist from keras.layers import Activation, Dense from keras.models import Sequential from keras import optimizers from keras.utils.np_utils import to_categorical import matplotlib.pyplot as plt (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"]) history = model.fit(X_train, y_train, verbose=True, epochs=3) #acc, val_accのプロット plt.plot(history.history["acc"], label="acc", ls="-", marker="o") plt.ylabel("accuracy") plt.xlabel("epoch") plt.legend(loc="best") plt.show()