schihei/simple-keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Uncommented compile of the model

Browse Source
This commit is contained in:
Heiko J Schick
2020-10-08 19:21:40 +02:00
parent e44eda9a9b
commit 8558acc4f6
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
prepare_train_model.py
+4 -5
View File
@@ -12,6 +12,7 @@ import matplotlib.pyplot as plt
import keras import keras
from keras.layers import Flatten, Dense from keras.layers import Flatten, Dense
from keras.applications.vgg16 import VGG16 from keras.applications.vgg16 import VGG16
from keras.optimizers import SGD
TRAIN_PATH = 'images/train/' TRAIN_PATH = 'images/train/'
TEST_PATH = 'images/test/' TEST_PATH = 'images/test/'
@@ -92,11 +93,9 @@ input("Press Enter to continue...")
#### Training #### Training
# Compile the model # Compile the model
# from keras.optimizers import SGD model.compile(loss='categorical_crossentropy',
optimizer=SGD(lr=1e-3),
# model.compile(loss='categorical_crossentropy', metrics=['accuracy'])
# optimizer=SGD(lr=1e-3),
# metrics=['accuracy'])
# Start the training process # Start the training process
# model.fit(x_train, y_train, validation_split=0.30, batch_size=32, epochs=50, verbose=2) # model.fit(x_train, y_train, validation_split=0.30, batch_size=32, epochs=50, verbose=2)