name: ium_s434766_mlflow_lab8

docker_env: my_env.yaml
    image:  s434732/ium:latest

entry_points:
  main:
    parameters:
        epochs = 5
        batch_size = 10
    command: "python mlflow.py -r {train_size_param} {test_size_param} {epochs} {batch_size}"