cellstrat-mlops-demo/MLproject

name: production-optimization-training

conda_env: conda.yaml

entry_points:
  main:
    parameters:
      max_epochs: {type: int, default: 50}
      batch_size: {type: int, default: 20}
      learning_rate: {type: float, default: 0.0003}
      alpha: {type: float, default: 0.1}

    command: |
          python training.py \
            --max_epochs {max_epochs} \
            --batch_size {batch_size} \
            --learning_rate {learning_rate} \
            --alpha {alpha}