tex-projects/LaTeX-examples
2
0
Fork 0
mirror of https://github.com/MartinThoma/LaTeX-examples.git synced 2025-04-26 06:48:04 +02:00
Code Activity

CNN Intro

Browse source
This commit is contained in:
Martin Thoma 2019-02-10 22:31:43 +01:00
parent 551102b9e7
commit 49b8552172
4 changed files with 108 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
presentations/CNN-Intro/cnn.py Normal file
View file

@ -0,0 +1,19 @@
import data
from keras.layers import Dense, Flatten, Conv2D, MaxPooling2D
from keras.models import Sequential, load_model
model = Sequential()
model.add(Conv2D(16, (3, 3)))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(16, (3, 3)))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(data.n_classes, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam')
model.fit(data.x_train, data.y_train)
model.save('model.h5')
model = load_model('model.h5')
y_predicted = model.predict(data.x_test)