BUG: Cast true labels to int in results dictionary (#682)

This commit is contained in:
Kenza Bouzid 2022-11-25 15:10:31 +00:00 коммит произвёл GitHub
Родитель 4f1e5f2859
Коммит 7c95a4f471
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -360,7 +360,7 @@ class DeepMILModule(LightningModule):
ResultsKey.PROB: predicted_probs,
ResultsKey.CLASS_PROBS: probs_perclass,
ResultsKey.PRED_LABEL: predicted_labels,
ResultsKey.TRUE_LABEL: bag_labels,
ResultsKey.TRUE_LABEL: bag_labels.int(),
ResultsKey.BAG_ATTN: bag_attn_list
})
self.update_results_with_metadata(batch=batch, results=results)