.

2022-04-25 03:53:55 +02:00 · 2022-04-25 03:53:55 +02:00 · e6b7def4b4
commit e6b7def4b4
parent 9ac404acff
1 changed files with 1 additions and 0 deletions
--- a/biblioteki_ml.py
+++ b/biblioteki_ml.py
@ -91,3 +91,4 @@ print(f"{df['Correct'].sum() / len(df)} percent of predictions correct")

 # Zapis do pliku
 df.to_csv('neural_network_prediction_results.csv', index=False)
+torch.save(model, "model.pkl")