diff --git a/Jenkinsfile_train-sacred b/Jenkinsfile_train-sacred index 4bed085..54a2bcb 100644 --- a/Jenkinsfile_train-sacred +++ b/Jenkinsfile_train-sacred @@ -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/**' diff --git a/predict.py b/predict.py index 4c26ba9..a64f8b8 100644 --- 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) \ No newline at end of file +print(f'Prediction: {loaded_model.predict(input_example)}') \ No newline at end of file