From bc74513edc48008ee0d6b1dfe84750974ce21169 Mon Sep 17 00:00:00 2001 From: Zofia Galla Date: Sat, 15 May 2021 14:31:31 +0200 Subject: [PATCH] Delete parameter learning_rate - Zadani 6 - training --- Jenkinsfile_training | 3 ++- ium_zadanie6_training.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Jenkinsfile_training b/Jenkinsfile_training index 99db88a..8755218 100644 --- a/Jenkinsfile_training +++ b/Jenkinsfile_training @@ -5,6 +5,7 @@ pipeline { defaultSelector: lastSuccessful(), description: 'Which build to use for copying artifacts', name: 'BUILD_SELECTOR') + string(name: 'LEARNING_RATE', defaultValue: '0.0001', description: 'What should be the learning rate?') } stages { stage('copyArtifacts') { @@ -15,7 +16,7 @@ pipeline { stage('Sh script') { steps { sh 'chmod +x run_training.sh' - sh './run_training.sh ${params.LEARNING_RATE}' + sh './run_training.sh' } diff --git a/ium_zadanie6_training.py b/ium_zadanie6_training.py index a1aad23..1b0b0e2 100644 --- a/ium_zadanie6_training.py +++ b/ium_zadanie6_training.py @@ -13,7 +13,7 @@ x_train = movies_train.copy() y_train = x_train.pop('rottentomatoes_audience_score') x_train.pop('Unnamed: 0') -learning_rate = sys.argv[1] +learning_rate = 0.001 #sys.argv[1] model = tf.keras.Sequential() model.add(layers.Input(shape=(22,)))