23 lines
480 B
Python
23 lines
480 B
Python
import csv
|
|
import pandas as pd
|
|
import seaborn as sns
|
|
import sys
|
|
import tensorflow
|
|
from tensorflow.keras import layers
|
|
from tensorflow.keras.models import load_model
|
|
|
|
|
|
X_test = pd.read_csv('test.csv')
|
|
|
|
Y_test = X_test.pop('stabf')
|
|
Y_test = pd.get_dummies(Y_test)
|
|
|
|
model = 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)
|
|
|