ium_z487179/ML/Jenkinsfile

42 lines
1.2 KiB
Groovy

pipeline {
agent any
parameters {
buildSelector(
name: 'BUILD_SELECTOR',
defaultSelector: lastSuccessful(),
description: 'A build to take the artifacts from'
),
string(
name: 'EPOCHS',
description: 'Number of epochs',
defaultValue: '10'
)
}
stages {
stage('Copy artifacts') {
steps {
script {
copyArtifacts(
projectName: 'z-s487179-create-dataset',
selector: buildParameter('BUILD_SELECTOR'),
target: './ML'
)
}
}
}
stage('Run training and save model') {
steps {
script {
sh 'ls -l'
docker.image('docker-image').inside {
dir('./ML') {
sh 'ls -l'
sh 'python3 ./model_train.py'
archiveArtifacts 'model.pt'
}
}
}
}
}
}
}