18 lines
385 B
Python
18 lines
385 B
Python
import csv
|
|
import pandas as pd
|
|
import tensorflow as tf
|
|
|
|
|
|
X_test = pd.read_csv('test.csv')
|
|
|
|
Y_test = X_test.pop('stabf')
|
|
Y_test = pd.get_dummies(Y_test)
|
|
|
|
model = tf.keras.models.load_model('grid-stability-dense.h5')
|
|
results = model.evaluate(X_test, Y_test, batch_size=64)
|
|
|
|
with open('eval.csv', 'a', newline='') as fp:
|
|
wr = csv.writer(fp, dialect='excel')
|
|
wr.writerow(results)
|
|
|