JoelJebaraj93 commited on
Commit
d9c50de
·
verified ·
1 Parent(s): 1b55e4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ model = Sequential([
22
  Dense(1)
23
  ])
24
  model.compile(optimizer="adam", loss="mse")
25
- model.fit(x, y, epochs=800, batch_size=512, verbose=0) # use 1 epoch for faster Hugging Face init
26
 
27
  # Prediction function
28
  def predict_next(n1, n2, n3):
 
22
  Dense(1)
23
  ])
24
  model.compile(optimizer="adam", loss="mse")
25
+ model.fit(x, y, epochs=1000, batch_size=512, verbose=0) # use 1 epoch for faster Hugging Face init
26
 
27
  # Prediction function
28
  def predict_next(n1, n2, n3):