Add learning_rate parameter
This commit is contained in:
parent
3344d3814d
commit
87737e9520
@ -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'
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
@ -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))
|
||||
|
Loading…
Reference in New Issue
Block a user