diff --git a/MLproject b/MLproject new file mode 100644 index 0000000..6c33695 --- /dev/null +++ b/MLproject @@ -0,0 +1,10 @@ +name: s478839 + +docker_env: + image: sebastianwalesa + +entry_points: + main: + parameters: + epochs: {type: float, default: 1000} + command: "python ml_pytorch_mlflow.py {epochs}"