mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-14 02:35:40 +00:00
29 lines
598 B
YAML
29 lines
598 B
YAML
$schema: https://componentsdk.azureedge.net/jsonschema/CommandComponent.json
|
|
# TODO: update name
|
|
name: classifier
|
|
version: 0.0.1
|
|
display_name: Train lgbm classifier
|
|
inputs:
|
|
train_data:
|
|
type: path
|
|
test_data:
|
|
type: path
|
|
learning_rate:
|
|
type: float
|
|
n_estimators:
|
|
type: int
|
|
outputs:
|
|
model:
|
|
type: path
|
|
environment:
|
|
conda:
|
|
conda_dependencies_file: env.yaml
|
|
os: Linux
|
|
command: >-
|
|
python train.py
|
|
--train_data {inputs.train_data}
|
|
--test_data {inputs.test_data}
|
|
--learning_rate {inputs.learning_rate}
|
|
--n_estimators {inputs.n_estimators}
|
|
--model {outputs.model}
|