aboutsummaryrefslogtreecommitdiff
path: root/lenet.py
diff options
context:
space:
mode:
Diffstat (limited to 'lenet.py')
-rw-r--r--lenet.py13
1 files changed, 0 insertions, 13 deletions
diff --git a/lenet.py b/lenet.py
index 39a5cd7..3d388de 100644
--- a/lenet.py
+++ b/lenet.py
@@ -83,19 +83,6 @@ def get_lenet_icp(shape):
model.add(Dense(units=10, activation = 'relu'))
return model
-def get_lenet_pen(shape):
- model = keras.Sequential()
- model.add(Conv2D(filters=6, kernel_size=(3, 3), activation='relu', input_shape=(32,32,1)))
- model.add(AveragePooling2D())
-
- model.add(Conv2D(filters=16, kernel_size=(3, 3), activation='relu'))
- model.add(AveragePooling2D())
- model.add(Flatten())
-
- model.add(Dense(units=120, activation='relu'))
- model.add(Dense(units=84, activation='relu'))
- return model
-
def plot_history(history, metric = None):
# Plots the loss history of training and validation (if existing)
# and a given metric