From fbcaa0b8618763989e92537030dc4b3763402981 Mon Sep 17 00:00:00 2001
From: Prashant Shekhar <50853959+prashant-shekhar@users.noreply.github.com>
Date: Thu, 17 Oct 2019 05:41:39 +0530
Subject: [PATCH] Updated Model Architecture

---
 Deep Learning/Dogs_vs_cats_CNN.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Deep Learning/Dogs_vs_cats_CNN.py b/Deep Learning/Dogs_vs_cats_CNN.py
index 73f0a7f..04240d4 100644
--- a/Deep Learning/Dogs_vs_cats_CNN.py	
+++ b/Deep Learning/Dogs_vs_cats_CNN.py	
@@ -127,6 +127,10 @@ def create_train_data():
 model.add(Activation("relu"))
 model.add(MaxPooling2D(pool_size=(2,2)))
 
+model.add(Conv2D(64, (3,3)))
+model.add(Activation("relu"))
+model.add(MaxPooling2D(pool_size=(2,2)))
+
 model.add(Flatten())
 model.add(Dense(64))
 model.add(Activation("relu"))
@@ -134,6 +138,8 @@ def create_train_data():
 model.add(Dense(1))
 model.add(Activation("sigmoid"))
 
+model.summary()
+
 model.compile(loss="binary_crossentropy",
              optimizer='adam',
              metrics=['accuracy'])