diff --git a/MLproject.txt b/MLproject.txt new file mode 100644 index 0000000..5bf82a3 --- /dev/null +++ b/MLproject.txt @@ -0,0 +1,16 @@ +name: s487197 + +docker_env: + image: s487197/ium:55 + +entry_points: + main: + parameters: + epochs: {type: int, default: 20} + lr: {type: float, default: 0.01} + validation_split: {type: float, default: 0.2} + command: "python ium_train.py {epochs} {lr} {validation_split}" + test: + parameters: + build: {type: int, default: 1} + command: "python ium_predict.py {build}" \ No newline at end of file