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

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 {