diff --git a/MLProject b/MLProject new file mode 100644 index 0000000..48a4f36 --- /dev/null +++ b/MLProject @@ -0,0 +1,12 @@ +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}"