s434784/ium_434784
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Archive model

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

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

2
training.py
View File

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