new impoer

This commit is contained in:
jakubknczny 2021-05-15 21:55:28 +02:00
parent 0ecc2edb39
commit b676105f9d

View File

@ -1,6 +1,6 @@
import csv import csv
import pandas as pd import pandas as pd
import tensorflow as tf from tensorflow.keras.models import load_model
X_test = pd.read_csv('test.csv') X_test = pd.read_csv('test.csv')
@ -8,7 +8,7 @@ X_test = pd.read_csv('test.csv')
Y_test = X_test.pop('stabf') Y_test = X_test.pop('stabf')
Y_test = pd.get_dummies(Y_test) Y_test = pd.get_dummies(Y_test)
model = tf.keras.models.load_model('grid-stability-dense.h5') model = load_model('grid-stability-dense.h5')
results = model.evaluate(X_test, Y_test, batch_size=64) results = model.evaluate(X_test, Y_test, batch_size=64)
with open('eval.csv', 'a', newline='') as fp: with open('eval.csv', 'a', newline='') as fp: