s434788/ium_434788
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fiiix

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-15 15:47:36 +02:00
parent 3c968505d3
commit 2e271479c3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Zadanie_06_evaluate.py
View File

@ -6,7 +6,7 @@ import sys
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from sklearn.metrics import accuracy_score, classification_report 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') print('evaluating')
test_df =pd.read_csv('test.csv') test_df =pd.read_csv('test.csv')

2
Zadanie_06_training.py
View File

@ -33,4 +33,4 @@ def regression_model():
model = 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.fit(x_train, y_train, batch_size=int(sys.argv[1]), epochs = int(sys.argv[1])) #verbose = 1
model.save('wine_model') model.save('wine_model.h5')