name: mlflow_464962

conda_env: conda.yaml

entry_points:
  main:
    parameters:
      epochs: {type: int, default: 20}
      batch_size: {type: int, default: 32}
    command: "python mlflow_model.py {epochs} {batch_size}"
  predict:
    command: "python mlflow_predict.py"