valohai.yaml

- step:
    name: autotune
    image: arimbr/valohai-fasttext-example
    command: python cli.py classification autotune {parameters}
    inputs:
      - name: train
      - name: validation
    parameters:
     - name: metric
       description: metric objective {f1, f1:labelname}
       type: string
       default: f1
     - name: k
       description: number of predictions used for evaluation
       type: integer
       default: 1
     - name: duration
       description: maximum duration in seconds
       type: integer
       default: 1200
     - name: model_size
       description: constraint model file size (empty = do not quantize)
       type: string
       default: 2000M