From b676105f9d0ee1cd78de96428ca9c0029502f777 Mon Sep 17 00:00:00 2001
From: jakubknczny <jakubknczny@gmail.com>
Date: Sat, 15 May 2021 21:55:28 +0200
Subject: [PATCH] new impoer

---
 lab5/eval/eval.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lab5/eval/eval.py b/lab5/eval/eval.py
index 5a20969..4e9f330 100644
--- a/lab5/eval/eval.py
+++ b/lab5/eval/eval.py
@@ -1,6 +1,6 @@
 import csv
 import pandas as pd
-import tensorflow as tf
+from tensorflow.keras.models import load_model
 
 
 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 = 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)
 
 with open('eval.csv', 'a', newline='') as fp: