From 02ca8d44657200264d6146538ab021f10f5e3d96 Mon Sep 17 00:00:00 2001
From: Jason Zerbe <jzerbe@vraidsys.com>
Date: Sun, 15 Nov 2020 10:41:24 -0700
Subject: [PATCH] labeled data output is the prediction concatenated on the
 historical

---
 src/index.js | 33 ++++++++++++++++++++-------------
 1 file changed, 20 insertions(+), 13 deletions(-)

diff --git a/src/index.js b/src/index.js
index 360bff7..df9e835 100644
--- 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) {