From 2e271479c32683166ec0d122f81aa6a062abf7e8 Mon Sep 17 00:00:00 2001
From: Dominik Strzako <sylvaripl@gmail.com>
Date: Sat, 15 May 2021 15:47:36 +0200
Subject: [PATCH] fiiix

---
 Zadanie_06_evaluate.py | 2 +-
 Zadanie_06_training.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Zadanie_06_evaluate.py b/Zadanie_06_evaluate.py
index 9d13e82..34aa474 100644
--- a/Zadanie_06_evaluate.py
+++ b/Zadanie_06_evaluate.py
@@ -6,7 +6,7 @@ import sys
 import matplotlib.pyplot as plt
 from sklearn.metrics import accuracy_score, classification_report
 
-model = keras.models.load_model('saved_model.pb')
+model = keras.models.load_model('wine_model.h5')
 print('evaluating')
 
 test_df =pd.read_csv('test.csv')
diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py
index 1997aa1..09bad05 100644
--- a/Zadanie_06_training.py
+++ b/Zadanie_06_training.py
@@ -33,4 +33,4 @@ def regression_model():
 model = regression_model()
 model.fit(x_train, y_train, batch_size=int(sys.argv[1]), epochs = int(sys.argv[1])) #verbose = 1
 
-model.save('wine_model')
\ No newline at end of file
+model.save('wine_model.h5')
\ No newline at end of file