IUM_06 - add epochs, learning_rate and weight_decay parameters models/Jenkinsfile

2024-04-19 11:08:18 +02:00 · 2024-04-19 11:08:18 +02:00 · b59420d33d
commit b59420d33d
parent 300d8a82d5
1 changed files with 15 additions and 1 deletions
--- a/models/Jenkinsfile
+++ b/models/Jenkinsfile
@ -2,7 +2,21 @@ pipeline {
    agent any

    parameters {
-
+        string(
+            name: 'epochs',
+            defaultValue: '1000',
+            description: 'Number of epochs for training'
+        )
+        string(
+            name: 'learning_rate',
+            defaultValue: '0.001',
+            description: 'Learning rate for training'
+        )
+        string(
+            name: 'weight_decay',
+            defaultValue: '0.001',
+            description: 'Regularization parameter for training'
+        )
    }

    stages {