name: s478841 regression model

docker_env:
  image: s478841-image:latest

entry_points:
  main:
    parameters:
      epochs: { type: string, default: "140" }
      steps: { type: string, default: "10" }
      save_model: { type: string, default: "--save" }
    command: "python3 scripts/mlflow_train.py -e {epochs} -s {steps} {save_model}"