Skip to content

Commit e563f62

Browse files
authored
Fix confusion matrix calculation (#3857)
* Fix confusion metrix calculation * Update accuracy.py
1 parent 34fdc7c commit e563f62

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/otx/api/usecases/evaluation/accuracy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ def __get_gt_and_predicted_label_indices_from_resultset(
218218
pred_dataset.sort_items()
219219

220220
# Iterate over each dataset item, and collect the labels for this item (pred and gt)
221-
task_labels = resultset.model.configuration.get_label_schema().get_labels(include_empty=True)
221+
task_labels = resultset.model.configuration.get_label_schema().get_labels(include_empty=False)
222222
for gt_item, pred_item in zip(gt_dataset, pred_dataset):
223223
if isinstance(gt_item, DatasetItemEntity) and isinstance(pred_item, DatasetItemEntity):
224224
true_label_idx.append({task_labels.index(label) for label in gt_item.get_roi_labels(task_labels)})

0 commit comments

Comments
 (0)