diff --git a/create_model.py b/create_model.py index af45a23..869c992 100644 --- a/create_model.py +++ b/create_model.py @@ -5,6 +5,7 @@ import torch.nn as nn import torch.optim as optim import os +import sys from sklearn.metrics import classification_report @@ -34,10 +35,18 @@ y_test = torch.FloatTensor(y_test).view(-1, 1) # Parameters input_size = X_train.shape[1] hidden_size = 128 + +# Default parameters learning_rate = 0.001 weight_decay = 0.001 num_epochs = 1000 +# Parameters from sys.argv +if len(sys.argv) > 1: + num_epochs = int(sys.argv[1]) + learning_rate = float(sys.argv[2]) + weight_decay = float(sys.argv[3]) + # Model initialization model = NeuralNetwork(input_size, hidden_size) diff --git a/models/Jenkinsfile b/models/Jenkinsfile index f534789..38f7b39 100644 --- a/models/Jenkinsfile +++ b/models/Jenkinsfile @@ -1,6 +1,10 @@ pipeline { agent any + parameters { + + } + stages { stage('Clone repository') { steps {