s434684/ium_434684
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add learning_rate parameter

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Jenkinsfile_training
View File

@ -16,8 +16,9 @@ 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' withEnv(["LEARNING_RATE=${params.LEARNING_RATE}"]) {
sh './run_training.sh LEARNING_RATE'
}
} }
} }

3
ium_zadanie6_training.py
View File

@ -13,8 +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 = 0.001 #sys.argv[1] learning_rate = float(sys.argv[1])
model = Sequential() model = Sequential()
model.add(layers.Input(shape=(22,))) model.add(layers.Input(shape=(22,)))
model.add(layers.Dense(64)) model.add(layers.Dense(64))