Update README.md

2026-06-27 17:01:51 +08:00 · 2022-08-02 09:32:58 -04:00
parent ccc35afb31
commit 81be06bcc1
1 changed files with 1 additions and 1 deletions
@@ -60,7 +60,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 estimator = DeepAREstimator(freq="5min",
                            prediction_length=12,
-                            input_size=43,
+                            input_size=19,
                            trainer=Trainer(epochs=10,
                                            device=device))
 predictor = estimator.train(training_data=training_data, num_workers=4)