Update model.py

vb100 · Mar 24, 2021 · 68232d4 · 68232d4
1 parent 04b91ca
commit 68232d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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