When processing raw data, if the task is classification and the outputs are not strings, convert them into strings. (#645)

t111 · joeyklee · commit e61339ade093 · 2019-10-18T17:35:31.000-04:00
diff --git a/src/NeuralNetwork/NeuralNetworkData.js b/src/NeuralNetwork/NeuralNetworkData.js
@@ -135,6 +135,10 @@ class NeuralNetworkData {
 
         if (outputLabels.includes(prop)) {
           output.ys[prop] = item[prop]
+          // convert ys into strings, if the task is classification
+          if(this.config.architecture.task === "classification" && typeof output.ys[prop] !== "string"){
+            output.ys[prop] += "";
+          }
         }
       })
 

Original file line number	Diff line number	Diff line change
`@@ -135,6 +135,10 @@ class NeuralNetworkData {`
`135`	`135`
`136`	`136`	`if (outputLabels.includes(prop)) {`
`137`	`137`	`output.ys[prop] = item[prop]`
	`138`	`+ // convert ys into strings, if the task is classification`
	`139`	`+ if(this.config.architecture.task === "classification" && typeof output.ys[prop] !== "string"){`
	`140`	`+ output.ys[prop] += "";`
	`141`	`+ }`
`138`	`142`	`}`
`139`	`143`	`})`
`140`	`144`