41 lines
1.0 KiB
Plaintext
41 lines
1.0 KiB
Plaintext
pipeline {
|
|
agent {
|
|
docker {
|
|
image 'python:3.8'
|
|
args '-v /root/.cache:/root/.cache -u root'
|
|
}
|
|
}
|
|
environment {
|
|
SACRED_IGNORE_GIT = 'TRUE'
|
|
}
|
|
parameters {
|
|
string(name: 'EPOCHS', defaultValue: '10', description: 'Liczba Epok')
|
|
}
|
|
stages {
|
|
stage('Preparation') {
|
|
steps {
|
|
sh 'pip install pandas tensorflow scikit-learn imbalanced-learn sacred pymongo'
|
|
}
|
|
}
|
|
stage('Pobierz dane') {
|
|
steps {
|
|
script {
|
|
copyArtifacts(projectName: 's487187-create-dataset', fingerprintArtifacts: true)
|
|
}
|
|
}
|
|
}
|
|
stage('Trenuj model') {
|
|
steps {
|
|
script {
|
|
sh "python3 train.py --epochs $EPOCHS"
|
|
}
|
|
}
|
|
}
|
|
stage('Zarchiwizuj model') {
|
|
steps {
|
|
archiveArtifacts artifacts: 'model.h5', fingerprint: true
|
|
}
|
|
}
|
|
}
|
|
}
|