diff --git a/lab5/eval/Jenkinsfile b/lab5/eval/Jenkinsfile index 82cc9ca..c7452df 100644 --- a/lab5/eval/Jenkinsfile +++ b/lab5/eval/Jenkinsfile @@ -13,10 +13,10 @@ pipeline { agent any steps { sh ''' - cp ./lab5/train/Dockerfile . - cp ./lab5/train/script.sh . - cp ./lab5/train/eval.py . - cp ./lab5/train/requirements.txt . + cp ./lab5/eval/Dockerfile . + cp ./lab5/eval/script.sh . + cp ./lab5/eval/eval.py . + cp ./lab5/eval/requirements.txt . ''' } } diff --git a/lab5/train/train.py b/lab5/train/train.py index a4780e2..0f3c519 100644 --- a/lab5/train/train.py +++ b/lab5/train/train.py @@ -1,4 +1,3 @@ -import numpy as np import pandas as pd import sys import tensorflow @@ -29,8 +28,6 @@ model.compile( optimizer=tensorflow.keras.optimizers.Adam(lr=float(sys.argv[1])), metrics=[tensorflow.keras.metrics.BinaryAccuracy()]) - -# history = model.fit(tensorflow.convert_to_tensor(X_train, np.float32), history = model.fit(X_train, Y_train, epochs=2, validation_data=(X_valid, Y_valid)) model.save('grid-stability-dense.h5')