eval the beginning

This commit is contained in:
jakubknczny 2021-05-15 19:35:54 +02:00
parent 7133025403
commit c393a0622c
2 changed files with 4 additions and 7 deletions

View File

@ -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 .
'''
}
}

View File

@ -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')