Update model.py

vb100 · Mar 24, 2021 · d775833 · d775833
1 parent 68232d4
commit d775833
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -84,7 +84,7 @@ def mse(y_test, y_pred):
               metrics = ['mae'])
 
 # Fit the model
-model.fit(X_train, y_train, epochs=100)
+model.fit(X_train, y_train, epochs=50)
 
 
 # Make and plot predictions for model_1