MLproject

@@ -1,13 +0,0 @@
-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():