-
Notifications
You must be signed in to change notification settings - Fork 1.3k
/
train.yml
35 lines (33 loc) · 844 Bytes
/
train.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
$schema: https://azuremlschemas.azureedge.net/latest/commandComponent.schema.json
name: my_train
display_name: Train_upper_case
#version: 1b
type: command
inputs:
training_data:
type: uri_folder
max_epocs:
type: integer
min: 0
max: 100
learning_rate:
type: number
default: 0.01
learning_rate_schedule:
type: string
default: time-based
enum:
- "step"
- "time-based"
outputs:
model_output:
type: uri_folder
code: ./train_src
environment: azureml://registries/azureml/environments/sklearn-1.0/labels/latest
command: >-
python train.py
--training_data ${{inputs.training_data}}
--max_epocs ${{inputs.max_epocs}}
--learning_rate ${{inputs.learning_rate}}
--learning_rate_schedule ${{inputs.learning_rate_schedule}}
--model_output ${{outputs.model_output}}