From ad29768ab67779414b4929035778229d4b082902 Mon Sep 17 00:00:00 2001 From: AdamOsiowy123 Date: Sun, 15 May 2022 23:07:36 +0200 Subject: [PATCH] mlflow fix4 --- Dockerfile | 3 ++- Jenkins/Jenkinsfile.training | 2 +- Scripts/train_neural_network.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 9937241..cf992bb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,8 @@ FROM ubuntu:latest RUN apt update && apt install -y python3 \ - python3-pip + python3-pip \ + git WORKDIR /code diff --git a/Jenkins/Jenkinsfile.training b/Jenkins/Jenkinsfile.training index 0e5f52d..7c60cce 100644 --- a/Jenkins/Jenkinsfile.training +++ b/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) { diff --git a/Scripts/train_neural_network.py b/Scripts/train_neural_network.py index 12189e7..1a00141 100644 --- a/Scripts/train_neural_network.py +++ b/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)