feat: create mlflow experiments monitoring

This commit is contained in:
Filip Patyk 2023-05-12 20:14:48 +02:00
parent 5fb057030a
commit fb944fa73e

13
MLproject Executable file
View File

@ -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"