Zaktualizuj 'biblioteka_DL'
This commit is contained in:
parent
a1429eea6e
commit
aff6e0827b
@ -73,4 +73,6 @@ with torch.no_grad():
|
|||||||
acc = y_predicted_cls.eq(y_testing).sum()/float(y_testing.shape[0])
|
acc = y_predicted_cls.eq(y_testing).sum()/float(y_testing.shape[0])
|
||||||
print(f'{acc:.4f}')
|
print(f'{acc:.4f}')
|
||||||
result = open("output",'w+')
|
result = open("output",'w+')
|
||||||
result.write(f'{y_predicted}')
|
result.write(f'{y_predicted}')
|
||||||
|
|
||||||
|
torch.save(model, "model.pkl")
|
Loading…
Reference in New Issue
Block a user