mlflow fix4

Dockerfile

@@ -1,7 +1,8 @@
 FROM ubuntu:latest
 
 RUN apt update && apt install -y python3 \
-    python3-pip
+    python3-pip \
+    git
 
 WORKDIR /code
 

Jenkins/Jenkinsfile.training

@@ -47,7 +47,7 @@ node {
         stage('Archive artifacts') {
             archiveArtifacts "model/neural_net"
             archiveArtifacts "my_runs/**"
-            archiveArtifacts "/code/mlruns/**"
+            archiveArtifacts "mlruns/**"
         }
         }
     } catch (e) {

Scripts/train_neural_network.py

@@ -19,7 +19,7 @@ import logging
 logging.basicConfig(level=logging.WARN)
 logger = logging.getLogger(__name__)
 
-# mlflow.set_tracking_uri("http://172.17.0.1:5000")
+mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment("s444452")
 
 ex = Experiment(name='s444452_fake_job_classification_training', save_git_info=False)