s434704/ium_434704
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Change training model params
Some checks failed
s434704-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-05-15 03:00:10 +02:00
parent 2a3a047b7a
commit 85b20518a4
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Jenkinsfile_train
View File

@ -4,7 +4,8 @@ pipeline {
}
parameters {
gitParameter branchFilter: 'origin/(.*)', defaultValue: 'master', name: 'BRANCH', type: 'PT_BRANCH'
string(name: 'options', description: 'Trainig script options')
string(name: 'epochs', description: 'Model epochs')
string(name: 'verbose', description: 'Model verbose')
buildSelector(defaultSelector: lastSuccessful(), description: 'Use latest build', name: 'BUILD_SELECTOR')
}
stages {
@ -15,7 +16,7 @@ pipeline {
}
stage("Run training"){
steps {
sh "python3 training.py ${params.options}"
sh "python3 training.py ${params.verbose}" ${params.epochs}"
}
}
stage('Save trained model files') {

5
training.py
View File

@ -11,9 +11,8 @@ from tensorflow.keras.layers.experimental import preprocessing
arguments = sys.argv[1:]
verbose = [command.split('=')[1] for command in arguments if command.split('=')[0] == 'verbose']
epochs = [command.split('=')[1] for command in arguments if command.split('=')[0] == 'epochs']
verbose = arguments[0]
epochs = arguments[1]
pd.set_option("display.max_columns", None)