Add learning_rate parameter

This commit is contained in:
Zofia Galla 2021-05-15 16:20:20 +02:00
parent 3344d3814d
commit 87737e9520
2 changed files with 4 additions and 4 deletions

View File

@ -16,8 +16,9 @@ pipeline {
stage('Sh script') {
steps {
sh 'chmod +x run_training.sh'
sh './run_training.sh'
withEnv(["LEARNING_RATE=${params.LEARNING_RATE}"]) {
sh './run_training.sh LEARNING_RATE'
}
}
}

View File

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