This commit is contained in:
Jakub Zaręba 2023-05-10 21:02:34 +02:00
parent 176f7aa4fe
commit 7bad1cc28f
2 changed files with 7 additions and 6 deletions

View File

@ -2,17 +2,16 @@ FROM ubuntu
RUN apt-get update && apt-get install -y python3 python3-pip unzip
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred
RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred pymongo
RUN apt-get install -y git
COPY train.py /app/train.py
COPY predictions.py /app/predictions.py
COPY data.csv /app/data.csv
# COPY test_data.csv /app/test_data.csv
WORKDIR /app
export SACRED_IGNORE_GIT=TRUE
RUN export SACRED_IGNORE_GIT=TRUE
RUN python3 train.py
RUN python3 predictions.py
CMD ["python3", "predictions.py"]

View File

@ -3,6 +3,9 @@ pipeline {
docker {
image 'python:3.11'
args '-v /root/.cache:/root/.cache -u root'
environment {
SACRED_IGNORE_GIT = 'TRUE'
}
}
}
parameters {
@ -24,8 +27,7 @@ pipeline {
stage('Trenuj model') {
steps {
script {
// sh "python3 train.py --epochs $EPOCHS"
sh "python3 train.py"
sh "python3 train.py --epochs $EPOCHS"
}
}
}