labeled data output is the prediction concatenated on the historical

Vraid-Systems · Nov 15, 2020 · 02ca8d4 · 02ca8d4
1 parent 1b6dc6c
commit 02ca8d4
Showing 1 changed file with 20 additions and 13 deletions.
diff --git a/src/index.js b/src/index.js
@@ -99,23 +99,30 @@ const main = async () => {
     }
 
     if (featureLabels) {
-        const inputDataRaw = fs.readFileSync(predictionFileLocation)
-        const inputData = JSON.parse(inputDataRaw)
+        const apiDataRaw = fs.readFileSync(apiDataFileLocation)
+        const apiData = JSON.parse(apiDataRaw)
 
-        const outputData = lodash.map(
-            inputData,
-            (inputDataBar) => {
-                const labeledObject = {}
+        const predictionDataRaw = fs.readFileSync(predictionFileLocation)
+        const predictionData = JSON.parse(predictionDataRaw)
 
-                for (let featureIndex = 0; featureIndex < featureLabels.length; featureIndex += 1) {
-                    labeledObject[featureLabels[featureIndex]] = inputDataBar[featureIndex + 1]
-                }
+        const labelData = (inputDataBar) => {
+            const labeledObject = {}
 
-                return [inputDataBar[0], labeledObject]
-            },
-        )
+            for (let featureIndex = 0; featureIndex < featureLabels.length; featureIndex += 1) {
+                labeledObject[featureLabels[featureIndex]] = inputDataBar[featureIndex + 1]
+            }
 
-        return writeObjectToOutputFile(outputData, labeledFileLocation)
+            return [inputDataBar[0], labeledObject]
+        }
+
+        const labeledApiData = lodash.map(apiData, labelData)
+
+        const labeledPredictionData = lodash.map(predictionData, labelData)
+
+        return writeObjectToOutputFile(
+            labeledApiData.concat(labeledPredictionData),
+            labeledFileLocation,
+        )
     }
 
     if (stockTicker) {