Skip to content

Commit 5624d72

Browse files
author
vanpelt
committed
Auto accuracy
1 parent 58f20f0 commit 5624d72

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

keras-transfer/dogcat-bottleneck.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def train_top_model():
8787
model.add(Dense(1, activation='sigmoid'))
8888

8989
model.compile(optimizer='rmsprop',
90-
loss='binary_crossentropy', metrics=['binary_accuracy'])
90+
loss='binary_crossentropy', metrics=['accuracy'])
9191

9292
class Images(Callback):
9393
def on_epoch_end(self, epoch, logs):

keras-transfer/dogcat-transfer-resnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
# and a very slow learning rate.
3636
model.compile(loss='binary_crossentropy',
3737
optimizer=optimizers.SGD(lr=1e-4, momentum=0.9),
38-
metrics=['binary_accuracy'])
38+
metrics=['accuracy'])
3939

4040
train_dir = "dogcat-data/train"
4141
val_dir = "dogcat-data/validation"

0 commit comments

Comments
 (0)