correctly label new prediction output format

Vraid-Systems · Nov 15, 2020 · 1b6dc6c · 1b6dc6c
1 parent e4d87f5
commit 1b6dc6c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -105,11 +105,10 @@ const main = async () => {
         const outputData = lodash.map(
             inputData,
             (inputDataBar) => {
-                const features = inputDataBar[1]
                 const labeledObject = {}
 
-                for (let featureIndex = 0; featureIndex < features.length; featureIndex += 1) {
-                    labeledObject[featureLabels[featureIndex]] = features[featureIndex]
+                for (let featureIndex = 0; featureIndex < featureLabels.length; featureIndex += 1) {
+                    labeledObject[featureLabels[featureIndex]] = inputDataBar[featureIndex + 1]
                 }
 
                 return [inputDataBar[0], labeledObject]