Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
arrizalamin committed Feb 16, 2018
1 parent 8698499 commit 2fb2bd0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ Keris uses keras-like [functional API](https://keras.io/getting-started/function
from keris.model import Model
from keris.layers import Input, Conv2D, LeakyReLU, MaxPool2D, Concatenate, Dropout, Dense
from keris.optimizers import Adam
from keris.losses import softmax_crossentropy
from keris.losses import categorical_softmax_crossentropy
from keris.callbacks import EarlyStopping, ModelCheckpoint

input_layer = Input(input_shape=(3, 60, 60), name='input')
Expand All @@ -31,7 +31,7 @@ callbacks = [EarlyStopping(patience=5),


model = Model(hidden)
model.compile(loss_fn=softmax_crossentropy, optimizer=optimizer)
model.compile(loss_fn=categorical_softmax_crossentropy, optimizer=optimizer)

model.fit(train_data, validation_data, epochs=100, batch_size=32, callbacks=callbacks)

Expand Down

0 comments on commit 2fb2bd0

Please sign in to comment.