diff --git a/Jenkinsfile-evaluation b/Jenkinsfile-evaluation index 6945237..24b0eb9 100644 --- a/Jenkinsfile-evaluation +++ b/Jenkinsfile-evaluation @@ -4,20 +4,13 @@ pipeline { additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} --build-arg CUTOFF=${params.CUTOFF} -t sebastianwalesa" } } - parameters { - string( - defaultValue: '1000', - description: 'Number of epochs', - name: 'EPOCHS', - trim: false - ) - } + stages { - stage('Train-stage'){ + stage('Evaluate'){ steps { copyArtifacts filter: '*', projectName: 's478839-create-dataset' - sh 'python3 ./ml_pytorch.py $EPOCHS' - archiveArtifacts artifacts: 'Model_xPosition.pkl', followSymlinks: false + sh 'python3 ./ml_pytorch_results.py' + archiveArtifacts artifacts: 'metrics.txt, MSE.txt, RMSplot.png', followSymlinks: false } } }