name: MLflow_s464937
conda_env: conda.yaml
entry_points:
  main:
    parameters:
      epochs: { type: int, default: 20 }
    command: 'python mlflow_model.py {epochs}'