Skip to content

Commit 0c6fa86

Browse files
add step size
1 parent e0ddd39 commit 0c6fa86

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

articles/machine-learning/how-to-auto-train-forecast.md

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,7 @@ In this case, the forecasting job _only_ searches over Exponential Smoothing and
399399
```python
400400
# Search over all model classes except Prophet
401401
forecasting_job.set_training(
402-
blocked_training_algorithms=["prophet"]
402+
blocked_training_algorithms=["Prophet"]
403403
)
404404
```
405405

@@ -1234,6 +1234,7 @@ inputs:
12341234
max_concurrency_per_node: 4
12351235
parallel_step_timeout_in_seconds: 3700
12361236
forecast_mode: rolling
1237+
forecast_step: 1
12371238
retrain_failed_model: False
12381239
12391240
# pipeline outputs
@@ -1268,7 +1269,7 @@ jobs:
12681269
max_concurrency_per_node: ${{parent.inputs.max_concurrency_per_node}}
12691270
parallel_step_timeout_in_seconds: ${{parent.inputs.parallel_step_timeout_in_seconds}}
12701271
forecast_mode: ${{parent.inputs.forecast_mode}}
1271-
forecast_step: 1
1272+
forecast_step: ${{parent.inputs.forecast_step}}
12721273
max_nodes: ${{parent.inputs.max_nodes}}
12731274
optional_train_metadata: ${{parent.jobs.mm_train_node.outputs.run_output}}
12741275
outputs:
@@ -1369,6 +1370,7 @@ Parameter|Description
13691370
**max_concurrency_per_node** | Number of AutoML processes to run on each node. Hence, the total concurrency of a HTS job is `max_nodes * max_concurrency_per_node`.
13701371
**parallel_step_timeout_in_seconds** | Many models component timeout given in number of seconds.
13711372
**forecast_mode** | Inference mode for model evaluation. Valid values are `"recursive"` and "`rolling`". See the [model evaluation article](concept-automl-forecasting-evaluation.md) for more information.
1373+
**forecast_step** | Step size for rolling forecast. See the [model evaluation article](concept-automl-forecasting-evaluation.md) for more information.
13721374

13731375
# [Python SDK](#tab/python)
13741376

@@ -1414,6 +1416,7 @@ def hts_train_evaluate_factory(
14141416
parallel_step_timeout_in_seconds=3700,
14151417
max_nodes=4,
14161418
forecast_mode="rolling",
1419+
forecast_step=1,
14171420
forecast_level="SKU",
14181421
allocation_method='proportions_of_historical_average'
14191422
):
@@ -1432,7 +1435,8 @@ def hts_train_evaluate_factory(
14321435
optional_train_metadata=hts_train.outputs.run_output,
14331436
forecast_level=forecast_level,
14341437
allocation_method=allocation_method,
1435-
forecast_mode=forecast_mode
1438+
forecast_mode=forecast_mode,
1439+
forecast_step=forecast_step
14361440
)
14371441
compute_metrics_node = compute_metrics_component(
14381442
task="tabular-forecasting",
@@ -1502,6 +1506,7 @@ inputs:
15021506
parallel_step_timeout_in_seconds: 3700
15031507
max_nodes: 4
15041508
forecast_mode: rolling
1509+
forecast_step: 1
15051510
allocation_method: proportions_of_historical_average
15061511
forecast_level: # forecast level
15071512
@@ -1537,7 +1542,7 @@ jobs:
15371542
max_concurrency_per_node: ${{parent.inputs.max_concurrency_per_node}}
15381543
parallel_step_timeout_in_seconds: ${{parent.inputs.parallel_step_timeout_in_seconds}}
15391544
forecast_mode: ${{parent.inputs.forecast_mode}}
1540-
forecast_step: 1
1545+
forecast_step: ${{parent.inputs.forecast_step}}
15411546
max_nodes: ${{parent.inputs.max_nodes}}
15421547
optional_train_metadata: ${{parent.jobs.hts_train_node.outputs.run_output}}
15431548
forecast_level: ${{parent.inputs.forecast_level}}

0 commit comments

Comments
 (0)