s444356/ium_444356
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update dllib-mlflow.py
All checks were successful
s444356-evaluation/pipeline/head This commit looks good
Details
s444356-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Maciej Czajka 2022-05-11 15:17:17 +02:00
parent 73bd0a68c7
commit e10b25c9c9
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Biblioteka_DL/dllib-mlflow.py
View File

@ -316,7 +316,7 @@ def my_main(epochs):
pred = pred.detach().numpy()
print("The accuracy is", accuracy_score(labels_test_g, np.argmax(pred, axis=1)))
# mlflow.log_metric("accuracy", accuracy_score(labels_test_g, np.argmax(pred, axis=1)))
mlflow.log_metric("accuracy", accuracy_score(labels_test_g, np.argmax(pred, axis=1)))
pred = pd.DataFrame(pred)
@ -324,11 +324,9 @@ def my_main(epochs):
# save model
torch.save(model, "games_model.pkl")
acc = accuracy_score(labels_test_g, np.argmax(pred, axis=1))
return acc
with mlflow.start_run() as run:
acc = my_main(epochs)
my_main(epochs)
mlflow.log_param("epochs", epochs)
mlflow.log_metric("accuracy", acc)