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 { parameters {
gitParameter branchFilter: 'origin/(.*)', defaultValue: 'master', name: 'BRANCH', type: 'PT_BRANCH' 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') buildSelector(defaultSelector: lastSuccessful(), description: 'Use latest build', name: 'BUILD_SELECTOR')
} }
stages { stages {
@ -15,7 +16,7 @@ pipeline {
} }
stage("Run training"){ stage("Run training"){
steps { steps {
sh "python3 training.py ${params.options}" sh "python3 training.py ${params.verbose}" ${params.epochs}"
} }
} }
stage('Save trained model files') { 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:] arguments = sys.argv[1:]
verbose = [command.split('=')[1] for command in arguments if command.split('=')[0] == 'verbose'] verbose = arguments[0]
epochs = [command.split('=')[1] for command in arguments if command.split('=')[0] == 'epochs'] epochs = arguments[1]
pd.set_option("display.max_columns", None) pd.set_option("display.max_columns", None)