module_mixins.py updated with tensor return

2020-05-21 12:33:36 +02:00
parent 5292b6d986
commit b529d130df
2 changed files with 2 additions and 2 deletions
--- a/util/module_mixins.py
+++ b/util/module_mixins.py
@ -100,7 +100,7 @@ class BaseValMixin:

            uar_score = sklearn.metrics.recall_score(y_true, y_pred, labels=[0, 1], average='macro',
                                                     sample_weight=None, zero_division='warn')
-
+            uar_score = torch.as_tensor(uar_score)
            summary_dict['log'].update({f'uar{ident}_score': uar_score})
        return summary_dict