update jenkinsfile_train-sacred, predict.py

2022-05-11 20:36:54 +02:00 · 2022-05-11 20:36:54 +02:00 · 3d2efac477
commit 3d2efac477
parent b16fce0beb
2 changed files with 2 additions and 1 deletions
--- a/1
+++ b/1
@ -17,6 +17,7 @@ pipeline {
 	stage('Script'){
 		steps {
                        copyArtifacts filter: '*', projectName: 's444356-create-dataset'
+			sh 'rm -rf my_model'
 			sh "python Biblioteka_DL/dllib-mlflow.py $EPOCHS"
 			archiveArtifacts artifacts: 'games_model.pkl'
 			archiveArtifacts artifacts: 'mlruns/**'
--- a/predict.py
+++ b/predict.py
@ -13,4 +13,4 @@ with open(f'{logged_model}/'+input) as f:
    data = json.load(f)
    input_example = np.array([data['inputs'][0]], dtype=np.float32)

-loaded_model.predict(input_example)
+print(f'Prediction: {loaded_model.predict(input_example)}')