Archive model

This commit is contained in:
Maciej Sobkowiak 2021-05-16 17:06:25 +02:00
parent ae6e29613e
commit d409867cd9
2 changed files with 4 additions and 1 deletions

3
.gitignore vendored
View File

@ -1,2 +1,3 @@
__pycache__
.ipynb_checkpoints
.ipynb_checkpoints
sucides_model.h5

View File

@ -90,3 +90,5 @@ test_predictions = model.predict(X_test).flatten()
predictions = model.predict(X_test)
pd.DataFrame(predictions).to_csv('results.csv')
model.save('sucides_model.h5')