Exemple #1
0
m = Dense(32, activation='relu')(m)
op = Dense(3, activation='softmax')(m)

model = Model(input=ip, output=op)

model.summary()

model.compile(loss='categorical_crossentropy',
              optimizer='adam',
              metrics=['accuracy'])

history4 = model.fit(train_X_ex3,
                     train_y,
                     epochs=100,
                     batch_size=32,
                     verbose=0,
                     validation_data=(test_X_ex3, test_y))

plt.plot(history4.history['accuracy'], label='Train Accuracy')
plt.plot(history4.history['val_accuracy'], label='Validation Accuracy')
plt.xlabel('Epochs')
plt.ylabel('Accuracy')
plt.legend()

corrects, wrongs = model.evaluate(train_X_ex3, train_y)
print("accuracy train: ", corrects / (corrects + wrongs))
corrects, wrongs = model.evaluate(test_X_ex3, test_y)
print("accuracy: test", corrects / (corrects + wrongs))

cm = model.confusion_matrix(train_X_ex3, train_y)