fit limit for confusion matrix in model eval

voxel51 · Jan 9, 2025 · e370d30 · e370d30
1 parent c5d3bf5
commit e370d30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/packages/core/src/plugins/SchemaIO/components/NativeModelEvaluationView/Evaluation.tsx b/app/packages/core/src/plugins/SchemaIO/components/NativeModelEvaluationView/Evaluation.tsx
@@ -1679,7 +1679,7 @@ function getMatrix(matrices, config, maskTargets, compareMaskTargets?) {
     return compareMaskTargets?.[c] || maskTargets?.[c] || c;
   });
   const noneIndex = originalClasses.indexOf(NONE_CLASS);
-  if (parsedLimit < originalClasses.length) {
+  if (parsedLimit < originalClasses.length && noneIndex > -1) {
     labels.push(
       compareMaskTargets?.[NONE_CLASS] ||
         maskTargets?.[NONE_CLASS] ||