fix default patchtst

Nixtla · Nov 7, 2023 · d5f5b92 · d5f5b92
1 parent ff1566c
commit d5f5b92
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/nbs/models.ipynb b/nbs/models.ipynb
@@ -1683,7 +1683,7 @@
     "        \"input_size_multiplier\": [1, 2, 3],\n",
     "        \"h\": None,\n",
     "        \"hidden_size\": tune.choice([16, 128, 256]),\n",
-    "        \"n_head\": tune.choice([4, 16]),\n",
+    "        \"n_heads\": tune.choice([4, 16]),\n",
     "        \"patch_len\": tune.choice([16, 24]),\n",
     "        \"learning_rate\": tune.loguniform(1e-4, 1e-1),\n",
     "        \"scaler_type\": tune.choice([None, 'robust', 'standard']),\n",

diff --git a/neuralforecast/auto.py b/neuralforecast/auto.py
@@ -985,7 +985,7 @@ class AutoPatchTST(BaseAuto):
         "input_size_multiplier": [1, 2, 3],
         "h": None,
         "hidden_size": tune.choice([16, 128, 256]),
-        "n_head": tune.choice([4, 16]),
+        "n_heads": tune.choice([4, 16]),
         "patch_len": tune.choice([16, 24]),
         "learning_rate": tune.loguniform(1e-4, 1e-1),
         "scaler_type": tune.choice([None, "robust", "standard"]),

diff --git a/neuralforecast/models/hint.py b/neuralforecast/models/hint.py
@@ -24,7 +24,7 @@ def get_bottomup_P(S: np.ndarray):
 
     **References:**<br>
     - [Orcutt, G.H., Watts, H.W., & Edwards, J.B.(1968). \"Data aggregation and information loss\". The American
-    Economic Review, 58 , 773{787)](http://www.jstor.org/stable/1815532).
+    Economic Review, 58 , 773(787)](http://www.jstor.org/stable/1815532).
     """
     n_series = len(S)
     n_agg = n_series - S.shape[1]