reorder metadata legend before results (#181)

* sort legend by onehot array * add comment
ml5js · Aug 6, 2024 · 4e02adf · 4e02adf
1 parent a88a3a1
commit 4e02adf
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/NeuralNetwork/index.js b/src/NeuralNetwork/index.js
@@ -1102,6 +1102,13 @@ class DiyNeuralNetwork {
     if (meta !== null) {
       const label = Object.keys(meta.outputs)[0];
       const vals = Object.entries(meta.outputs[label].legend);
+      // sort the legend array results by each item's one-hot encoding
+      vals.sort((a, b) => {
+        return (
+          b[1].reduce((acc, curr) => acc * 2 + curr, 0) -
+          a[1].reduce((acc, curr) => acc * 2 + curr, 0)
+        );
+      });
 
       const formattedResults = unformattedResults.map((unformattedResult) => {
         return vals