Daniel Varga commited on
Commit
1fee7d7
1 Parent(s): 41bce18
Files changed (2) hide show
  1. v2/readme.sh +26 -0
  2. v2/test_predictor_statsforecast.py +2 -2
v2/readme.sh ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
2
+ bash Miniconda3-latest-Linux-x86_64.sh
3
+ conda install mamba -n base -c conda-forge
4
+ # gluonts does not like python 3.12.
5
+ mamba create -n py311 python=3.11
6
+ conda activate py311
7
+ mamba install matplotlib
8
+ mamba install "gluonts[torch]"
9
+ # gluonts does not like pandas 2.2 deprecating capital letter frequency strings. fixed on gluonts dev branch, but not on conda yet.
10
+ mamba install pandas=2.1
11
+ # for neuralforecast
12
+ mamba install -c conda-forge neuralforecast
13
+ # https://github.com/apache/arrow/issues/41058
14
+ mamba install snappy=1.1.10
15
+
16
+ python test_predictor_statsforecast.py
17
+ # do not use it with AutoARIMA(4*24), that's too slow.
18
+
19
+ # with EXTENDED_TEST = True, train_df before '2019-02-01', in 1.5 mins:
20
+ MSTL RMSE 30.83026 MAE 23.066748
21
+ SeasonalNaive RMSE 48.91051 MAE 36.31397
22
+ AutoARIMA RMSE 71.8839 MAE 55.241413
23
+ AutoETS RMSE 71.732056 MAE 55.39218
24
+ CES RMSE 80.47097 MAE 70.04802
25
+ DynamicOptimizedTheta RMSE 55.568027 MAE 43.394047
26
+
v2/test_predictor_statsforecast.py CHANGED
@@ -110,7 +110,7 @@ models = [
110
  SeasonalNaive(season_length=seasonality)
111
  ]
112
 
113
- EXTENDED_TEST = False
114
  if EXTENDED_TEST:
115
  models += [
116
  # AutoARIMA(season_length=4 * 24) is just too slow, never even finishes,
@@ -127,7 +127,7 @@ freq = '15min'
127
  sf = StatsForecast(
128
  models=models,
129
  freq=freq,
130
- n_jobs=1,
131
  )
132
 
133
 
 
110
  SeasonalNaive(season_length=seasonality)
111
  ]
112
 
113
+ EXTENDED_TEST = True
114
  if EXTENDED_TEST:
115
  models += [
116
  # AutoARIMA(season_length=4 * 24) is just too slow, never even finishes,
 
127
  sf = StatsForecast(
128
  models=models,
129
  freq=freq,
130
+ n_jobs=-1,
131
  )
132
 
133