name: MLflow_s464979

conda_env: conda.yaml

entry_points:
  optimal_parameters:
    parameters:
      epochs: { type: int, default: 20 }
      batch_size: { type: int, default: 32 }
    command: 'python mlflow_training_model.py {epochs} {batch_size}'