MLproject

name: malaria

docker_env:
  image: malaria
  volumes: ["/storage:/storage", "~/.pgpass:/root/.pgpass"]  # contains both data and mlruns
  # environment: [["NEW_ENV_VAR", "new_var_value"], "VAR_TO_COPY_FROM_HOST_ENVIRONMENT"]

entry_points:
  main:
    parameters:
      batch_size: {type: int, default: 32}
      epochs: {type: int, default: 10}
      convolutions: {type: int, default: 1}
      training_samples: {type: int}
      validation_samples: {type: int}
      randomize_images: {type: bool}
      run_name: {type: str, default: py_tf2_gpu}
    command: "python train.py \
      --batch-size {batch_size} \
      --epochs {epochs} \
      --convolutions {convolutions} \
      --training-samples {training_samples} \
      --validation-samples {validation_samples} \
      --randomize-images {randomize_images} \
      --run-name {run_name}"