From beadfe027df045e900cd5730b534402eba9a3113 Mon Sep 17 00:00:00 2001 From: Maciej Sobkowiak Date: Sun, 16 May 2021 22:19:30 +0200 Subject: [PATCH] . --- evaluation.py | 2 +- training.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/evaluation.py b/evaluation.py index b0673b5..12ac135 100644 --- a/evaluation.py +++ b/evaluation.py @@ -5,7 +5,7 @@ import matplotlib.pyplot as plt from keras import backend as K from sklearn.metrics import mean_squared_error -model = keras.models.load_model('suicide_model') +model = keras.models.load_model('suicide_model.h5') # train = pd.read_csv('train.csv') # test = pd.read_csv('test.csv') diff --git a/training.py b/training.py index ac0ba8e..8ecbbb1 100644 --- a/training.py +++ b/training.py @@ -91,4 +91,4 @@ test_predictions = model.predict(X_test).flatten() predictions = model.predict(X_test) pd.DataFrame(predictions).to_csv('results.csv') -model.save('suicide_model') +model.save('suicide_model.h5')