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 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 RUN apt-get install -y git
COPY train.py /app/train.py COPY train.py /app/train.py
COPY predictions.py /app/predictions.py COPY predictions.py /app/predictions.py
COPY data.csv /app/data.csv COPY data.csv /app/data.csv
# COPY test_data.csv /app/test_data.csv
WORKDIR /app WORKDIR /app
export SACRED_IGNORE_GIT=TRUE RUN export SACRED_IGNORE_GIT=TRUE
RUN python3 train.py RUN python3 train.py
RUN python3 predictions.py CMD ["python3", "predictions.py"]

View File

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