feat: create mlflow experiments monitoring

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"

src/main.py

@@ -28,7 +28,7 @@ parser.add_argument("--num_epochs", "--epochs", "-e", type=int, default=3)
 
 # mlflow stuff
 mlflow.set_tracking_uri("http://localhost:5000")
-mlflow.set_experiment("s424714")
+# mlflow.set_experiment("s424714")
 
 
 def main():