save results

2021-05-16 22:58:56 +02:00 · 2021-05-16 22:58:56 +02:00 · 7c428075a4
commit 7c428075a4
parent ba432415fb
2 changed files with 9 additions and 5 deletions
--- a/eval.Jenkinsfile
+++ b/eval.Jenkinsfile
@ -31,5 +31,10 @@ pipeline {
                }
 			}
 		}
+        stage('archive-artifacts') {
+			steps{
+				archiveArtifacts 'results.txt'
+            }
+        }
    }
 }
--- a/evaluation.py
+++ b/evaluation.py
@ -30,10 +30,9 @@ model.summary()

 model.load_weights('suicide_model.h5')

+predictions = model.predict(X_test)

-# predictions = model.predict(X_test)
+error = mean_squared_error(y_test, predictions)

-# error = mean_squared_error(y_test, predictions)
-
-# with open('eval_results.txt', 'a') as f:
-#     f.write(str(error) + "\n")
+with open('results.txt', 'a') as f:
+    f.write(str(error) + "\n")