new impoer
This commit is contained in:
parent
0ecc2edb39
commit
b676105f9d
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user