Zaktualizuj 'biblioteka_DL'

2022-05-06 19:21:59 +02:00 · 2022-05-06 19:21:59 +02:00 · aff6e0827b
commit aff6e0827b
parent a1429eea6e
1 changed files with 3 additions and 1 deletions
--- a/4
+++ b/4
@ -73,4 +73,6 @@ with torch.no_grad():
    acc = y_predicted_cls.eq(y_testing).sum()/float(y_testing.shape[0])
    print(f'{acc:.4f}')
    result = open("output",'w+')
-    result.write(f'{y_predicted}')
+    result.write(f'{y_predicted}')
+
+torch.save(model, "model.pkl")