feat: accuracy metric for int and category columns

Renumics · Sep 14, 2023 · 4af81f1 · 4af81f1
1 parent 7bc31e1
commit 4af81f1
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/src/widgets/MetricsWidget/metrics.ts b/src/widgets/MetricsWidget/metrics.ts
@@ -35,21 +35,23 @@ export const METRICS: Record<string, Metric> = {
     },
     accuracy: {
         signature: {
-            X: ['bool'],
-            y: ['bool'],
+            X: ['bool', 'int', 'Category'],
+            y: ['bool', 'int', 'Category'],
         },
         compute: ([actualValues, assignedValues]) => {
-            const {
-                truePositives: tp,
-                trueNegatives: tn,
-                falsePositives: fp,
-                falseNegatives: fn,
-            } = computeConfusion(
-                actualValues as boolean[],
-                assignedValues as boolean[]
-            );
+            const all = actualValues.length;
+            let correct = 0;
+
+            for (let i = 0; i < actualValues.length; i++) {
+                const actual = actualValues[i];
+                const assigned = assignedValues[i];
+
+                if (actual === assigned) {
+                    correct++;
+                }
+            }
 
-            return (tp + tn) / (tp + tn + fp + fn);
+            return correct / all;
         },
     },
     F1: {