From 68232d47cf0ac67b9eaaf62c4d0aa9d01dc7287d Mon Sep 17 00:00:00 2001
From: Vytautas Bielinskas <vytautas.bielinskas@gmail.com>
Date: Wed, 24 Mar 2021 14:07:43 +0200
Subject: [PATCH] Update model.py

---
 model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model.py b/model.py
index d37cf79..31ef863 100644
--- a/model.py
+++ b/model.py
@@ -75,7 +75,7 @@ def mse(y_test, y_pred):
 # Create a model using the Sequential API
 model = tf.keras.Sequential([
     tf.keras.layers.Dense(1), 
-    #tf.keras.layers.Dense(1)
+    tf.keras.layers.Dense(1)
     ])
 
 # Compile the model
@@ -84,7 +84,7 @@ def mse(y_test, y_pred):
               metrics = ['mae'])
 
 # Fit the model
-model.fit(X_train, y_train, epochs=50)
+model.fit(X_train, y_train, epochs=100)
 
 
 # Make and plot predictions for model_1