2023-05-10 15:11:42 +02:00
|
|
|
pipeline {
|
|
|
|
agent {
|
|
|
|
docker {
|
|
|
|
image 'python:3.11'
|
2023-05-10 16:57:40 +02:00
|
|
|
args '-v /root/.cache:/root/.cache -u root'
|
2023-05-10 15:11:42 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
parameters {
|
2023-05-10 15:36:16 +02:00
|
|
|
string(name: 'EPOCHS', defaultValue: '10', description: 'Liczba Epok')
|
2023-05-10 15:11:42 +02:00
|
|
|
}
|
|
|
|
stages {
|
2023-05-10 16:57:40 +02:00
|
|
|
stage('Preparation') {
|
|
|
|
steps {
|
2023-05-10 17:08:50 +02:00
|
|
|
sh 'pip install pandas tensorflow scikit-learn imbalanced-learn'
|
2023-05-10 16:57:40 +02:00
|
|
|
}
|
|
|
|
}
|
2023-05-10 15:11:42 +02:00
|
|
|
stage('Pobierz dane') {
|
|
|
|
steps {
|
|
|
|
script {
|
2023-05-10 16:46:45 +02:00
|
|
|
copyArtifacts(projectName: 's487187-create-dataset', fingerprintArtifacts: true)
|
2023-05-10 15:11:42 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
stage('Trenuj model') {
|
|
|
|
steps {
|
|
|
|
script {
|
2023-05-10 17:04:28 +02:00
|
|
|
// sh "python3 train.py --epochs $EPOCHS"
|
|
|
|
sh "python3 train.py"
|
2023-05-10 15:11:42 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
stage('Zarchiwizuj model') {
|
|
|
|
steps {
|
|
|
|
archiveArtifacts artifacts: 'model.h5', fingerprint: true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-05-10 17:08:50 +02:00
|
|
|
}
|