diff --git a/Dockerfile b/Dockerfile index da19e61..090a36d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -33,3 +33,5 @@ COPY ./plot.py ./ RUN chmod +x plot.py COPY ./train_mlflow.py ./ RUN chmod +x train_mlflow.py +COPY ./generate_MLmodel.py ./ +RUN chmod +x generate_MLmodel.py diff --git a/Jenkinsfile_train b/Jenkinsfile_train index 80d600b..10ed336 100644 --- a/Jenkinsfile_train +++ b/Jenkinsfile_train @@ -34,6 +34,8 @@ pipeline { sh 'chmod +x dlgssdpytorch.py' sh 'python3 ./dlgssdpytorch.py $PARAMETRY' sh 'chmod +x train_mlflow.py' + sh 'chmod +x generate_MLmodel.py' + sh 'python3 ./generate_MLmodel' //sh 'mlflow ./train_mlflow.py' //sh 'mlflow run .' }