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,)))