зеркало из https://github.com/microsoft/hi-ml.git
BUG: Cast true labels to int in results dictionary (#682)
This commit is contained in:
Родитель
4f1e5f2859
Коммит
7c95a4f471
|
@ -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)
|
||||
|
|
Загрузка…
Ссылка в новой задаче