How to use keras for binary classification?

Question

I need simple example about how to use keras model. It is not clear for me what difference between model.evaluate and model.predict.

I want to create model for binary classification. Lets say I have images of cats and dogs, train model and can use it to predict which animal on given photo. Maybe there is some good into or tutorials. I read anything in first five pages in google, but found only complex level tutorials and discussions.

nemo · Accepted Answer · 2016-09-29 20:28:44Z

8

To make things short:

model.evaluate evaluates a pair (X,Y) and returns the loss (and all other metrics configured for the model). This is for testing your model on a vaildation or test set.
model.predict predicts the outcome given an input X. This if for predicting the class from an input image for example.

This, among other things, is also clearly documented in the linked documentation.

You can find a lot of example models for Keras in the git repository (keras/examples) or on the Keras website (here and here).

For binary classification you could use this model for example:

model = Sequential()
model.add(Dense(300, init='uniform'))
model.add(Activation('relu'))
model.add(Dense(2, init='uniform'))
model.add(Activation('softmax'))

model.compile(loss='categorical_crossentropy', optimizer=Adam(lr=0.02))
model.fit(X, Y)

answered Sep 29, 2016 at 20:28

nemo

58.3k15 gold badges144 silver badges145 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to use keras for binary classification?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related