Skip to content

Commit

Permalink
Add NeuralForecast to evaluate_baseline.
Browse files Browse the repository at this point in the history
  • Loading branch information
fsaad committed May 24, 2024
1 parent 934e879 commit 56d7d7d
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion scripts/README
Original file line number Diff line number Diff line change
Expand Up @@ -51,14 +51,22 @@ Step 3. Run the baseline evaluation scripts.
flags:

evaluate_baseline.py:
--algorithm: <SVGP|ST-SVGP|MF-ST-SVGP|RF|GBOOST|TSREG>: Algorithm name
--algorithm: <SVGP|ST-SVGP|MF-ST-SVGP|RF|GBOOST|TSREG|NF>: Algorithm name
--data_root: Location of input data.
--dataset: <air_quality|wind|air|chickenpox|coprecip|sst>: Dataset name
--gboost_estimators: Number of GBOOST estimators.
(default: '100')
(an integer)
--[no]gboost_featurize: Add Fourier features to GBOOST baseline.
(default: 'false')
--nf_epochs: Number of epochs for NeuralForecast training.
(default: '5000')
(an integer)
--nf_method: Method for NeuralForecast baseline (https://nixtlaverse.nixtla.io/neuralforecast/models.html).
(default: 'NBEATS')
--nf_window: Number of previous horizons in NeuralForecast window.
(default: '5')
(an integer)
--output_dir: Output directory.
--start_id: Run experiments on series with IDs >= this value.
(default: '5')
Expand Down

0 comments on commit 56d7d7d

Please sign in to comment.