Skip to content

Commit 1b506f9

Browse files
committed
sobol example
1 parent e1ca57f commit 1b506f9

File tree

1 file changed

+44
-1
lines changed

1 file changed

+44
-1
lines changed

articles/machine-learning/reference-yaml-job-sweep.md

Lines changed: 44 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ ms.topic: reference
99
ms.custom: cliv2, devx-track-python, update-code6
1010
ms.author: franksolomon
1111
author: fbsolo-ms1
12-
ms.date: 03/05/2024
12+
ms.date: 12/03/2024
1313
ms.reviewer: amipatel
1414
---
1515

@@ -242,6 +242,49 @@ Visit the [examples GitHub repository](https://github.com/Azure/azureml-examples
242242

243243
:::code language="yaml" source="~/azureml-examples-main/cli/jobs/basics/hello-sweep.yml":::
244244

245+
## YAML: hello sweep using sobol and a seed value
246+
247+
```yml
248+
$schema: https://azuremlschemas.azureedge.net/latest/sweepJob.schema.json
249+
type: sweep
250+
trial:
251+
code: src
252+
command: >-
253+
python main.py
254+
--iris-csv ${{inputs.iris_csv}}
255+
--learning-rate ${{search_space.learning_rate}}
256+
--boosting ${{search_space.boosting}}
257+
environment: azureml:AzureML-lightgbm-3.3@latest
258+
inputs:
259+
iris_csv:
260+
type: uri_file
261+
path: https://azuremlexamples.blob.core.windows.net/datasets/iris.csv
262+
compute: azureml:cpu-cluster
263+
sampling_algorithm:
264+
type: random
265+
rule: sobol
266+
seed: 123
267+
search_space:
268+
learning_rate:
269+
type: uniform
270+
min_value: 0.01
271+
max_value: 0.9
272+
boosting:
273+
type: choice
274+
values: ["gbdt", "dart"]
275+
objective:
276+
goal: minimize
277+
primary_metric: test-multi_logloss
278+
limits:
279+
max_total_trials: 20
280+
max_concurrent_trials: 10
281+
timeout: 7200
282+
display_name: lightgbm-iris-sweep-example
283+
experiment_name: lightgbm-iris-sweep-example
284+
description: Run a hyperparameter sweep job for LightGBM on Iris dataset.
285+
```
286+
287+
245288
## YAML: basic Python model hyperparameter tuning
246289
247290
:::code language="yaml" source="~/azureml-examples-main/cli/jobs/single-step/scikit-learn/iris/job-sweep.yml":::

0 commit comments

Comments
 (0)