diff --git a/MLproject b/MLproject new file mode 100755 index 0000000..60e7b07 --- /dev/null +++ b/MLproject @@ -0,0 +1,13 @@ +name: s424714 + +conda_env: environment.yml #ścieżka do pliku conda.yaml z definicją środowiska + +entry_points: + main: + parameters: + epochs: {type: int, default: 3} + lr: {type: float, default: 0.000001} + command: "python src/main.py --train --lr={lr} -e={epochs}" + # command: "python src/train.py --train --lr={lr} -e={epochs}" + validate: + command: "python src/main.py --train" \ No newline at end of file