85 lines
2.2 KiB
Groovy
85 lines
2.2 KiB
Groovy
pipeline {
|
|
agent {
|
|
dockerfile true
|
|
}
|
|
|
|
parameters {
|
|
string(
|
|
defaultValue: '10000',
|
|
name: 'CUTOFF',
|
|
description: 'Liczba wierszy do obcięcia ze zbioru danych')
|
|
string(
|
|
defaultValue: '--epochs 100 --batch_size 32 --learning_rate 0.01',
|
|
name: 'TRAINING_PARAMS',
|
|
description: 'Parametry trenowania'
|
|
)
|
|
}
|
|
|
|
environment {
|
|
DATASET_PROJECT = 's123456-create-dataset'
|
|
DATA_DIR = 'data'
|
|
}
|
|
|
|
stages {
|
|
stage('Clone repository') {
|
|
steps {
|
|
checkout([$class: 'GitSCM', branches: [[name: '*/master']], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s464906/ium_464906']]])
|
|
}
|
|
}
|
|
stage('Copy Dataset') {
|
|
steps {
|
|
script {
|
|
copyArtifacts(
|
|
projectName: "${DATASET_PROJECT}",
|
|
selector: lastSuccessful(),
|
|
target: "${env.DATA_DIR}"
|
|
)
|
|
}
|
|
}
|
|
}
|
|
stage('Set execute permission') {
|
|
steps {
|
|
script {
|
|
sh 'chmod +x data_processing.sh'
|
|
}
|
|
}
|
|
}
|
|
stage('Run shell script') {
|
|
steps {
|
|
script {
|
|
sh './data_processing.sh'
|
|
}
|
|
}
|
|
post {
|
|
success {
|
|
archiveArtifacts artifacts: 'results.txt', allowEmptyArchive: true
|
|
}
|
|
}
|
|
}
|
|
|
|
stage('Install Dependencies') {
|
|
steps {
|
|
sh 'pip install -r requirements.txt'
|
|
}
|
|
}
|
|
|
|
stage('Train Model') {
|
|
steps {
|
|
sh "python model.py ${params.TRAINING_PARAMS}"
|
|
}
|
|
}
|
|
|
|
stage('Archive Model') {
|
|
steps {
|
|
archiveArtifacts artifacts: 'orange_quality_model_tf.h5', allowEmptyArchive: true
|
|
archiveArtifacts artifacts: 'predictions_tf.json', allowEmptyArchive: true
|
|
}
|
|
}
|
|
}
|
|
post {
|
|
always {
|
|
cleanWs()
|
|
}
|
|
}
|
|
}
|