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

View File

@ -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')

View File

@ -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')
model.save('wine_model.h5')