From e1827f7dddf600c181257e9d42ed05034f48d9db Mon Sep 17 00:00:00 2001
From: s444417 <mikolaj12341230@gmail.com>
Date: Sat, 14 May 2022 12:12:20 +0200
Subject: [PATCH] add uri

---
 Jenkinsfile3        |  5 +++++
 lab8/trainScript.py | 18 +++++++++---------
 result.txt          |  4 ++--
 3 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/Jenkinsfile3 b/Jenkinsfile3
index 7fa1375..0ca5507 100644
--- a/Jenkinsfile3
+++ b/Jenkinsfile3
@@ -33,6 +33,11 @@ pipeline {
                 build job: 's444417-evaluation/master', wait: false
             }
         }
+        stage('Archive mlflow') {
+            steps {
+                sh 'cd ./lab8'
+            }
+        }
     }
     options {
         copyArtifactPermission('s444417-evaluation');
diff --git a/lab8/trainScript.py b/lab8/trainScript.py
index c037f91..51dbb21 100644
--- a/lab8/trainScript.py
+++ b/lab8/trainScript.py
@@ -10,7 +10,7 @@ import mlflow.keras
 from urllib.parse import urlparse
 
 
-# mlflow.set_tracking_uri('http://localhost:5000')
+mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment('s444417')
 
 # train params
@@ -108,14 +108,14 @@ def train():
     mlflow.log_param('learning rate', learning_rate)
     mlflow.log_metric('val loss', min(hist["val_loss"]))
 
-  # signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))
-  #
-  # tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
-  #
-  # if tracking_url_type_store != "file":
-  #     mlflow.keras.log_model(linear_model, "linear-model", registered_model_name="HousePriceLinear", signature=signature)
-  # else:
-  #     mlflow.keras.log_model(linear_model, "model", signature=signature)
+  signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))
+  
+  tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
+  
+  if tracking_url_type_store != "file":
+      mlflow.keras.log_model(linear_model, "linear-model", registered_model_name="HousePriceLinear", signature=signature)
+  else:
+      mlflow.keras.log_model(linear_model, "model", signature=signature)
 
 if __name__ == '__main__':
     train()
\ No newline at end of file
diff --git a/result.txt b/result.txt
index 7d73d0d..c4c57b5 100644
--- a/result.txt
+++ b/result.txt
@@ -1,2 +1,2 @@
-predictions: [157.08437, 4.671051, 190.45694, 126.68617, 96.37216, 134.32784, 154.44032, -19.104736, 80.28882, 100.09464]
-expected: [110.   25.2  60.   70.  100.  110.  520.   18.   77.   32. ]
\ No newline at end of file
+predictions: [188.40611, 79.26458, 79.07924, 79.702156, 249.17542, -1.7695923, 54.21414, 78.15069, 114.4659, 248.62257]
+expected: [ 74. 140.  35.  42. 260.  15.  25. 190.  40. 200.]
\ No newline at end of file