Spaces:

kashif
/

probabilistic-forecast

Runtime error

Kashif Rasul commited on Apr 27, 2023

Commit

c029c40

1 Parent(s): 46a14b8

more epochs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ def fn(upload_data):
     model = DeepAREstimator(
         prediction_length=12,
         freq=dataset.freq,
-        trainer_kwargs=dict(max_epochs=1),
     ).train(
         training_data=training_data,
     )

     model = DeepAREstimator(
         prediction_length=12,
         freq=dataset.freq,
+        trainer_kwargs=dict(max_epochs=10),
     ).train(
         training_data=training_data,
     )