Delete parameter learning_rate - Zadani 6 - training

This commit is contained in:
Zofia Galla 2021-05-15 14:31:31 +02:00
parent 6a4fba089e
commit bc74513edc
2 changed files with 3 additions and 2 deletions

View File

@ -5,6 +5,7 @@ pipeline {
defaultSelector: lastSuccessful(), defaultSelector: lastSuccessful(),
description: 'Which build to use for copying artifacts', description: 'Which build to use for copying artifacts',
name: 'BUILD_SELECTOR') name: 'BUILD_SELECTOR')
string(name: 'LEARNING_RATE', defaultValue: '0.0001', description: 'What should be the learning rate?')
} }
stages { stages {
stage('copyArtifacts') { stage('copyArtifacts') {
@ -15,7 +16,7 @@ pipeline {
stage('Sh script') { stage('Sh script') {
steps { steps {
sh 'chmod +x run_training.sh' sh 'chmod +x run_training.sh'
sh './run_training.sh ${params.LEARNING_RATE}' sh './run_training.sh'
} }

View File

@ -13,7 +13,7 @@ x_train = movies_train.copy()
y_train = x_train.pop('rottentomatoes_audience_score') y_train = x_train.pop('rottentomatoes_audience_score')
x_train.pop('Unnamed: 0') x_train.pop('Unnamed: 0')
learning_rate = sys.argv[1] learning_rate = 0.001 #sys.argv[1]
model = tf.keras.Sequential() model = tf.keras.Sequential()
model.add(layers.Input(shape=(22,))) model.add(layers.Input(shape=(22,)))