testing

Signed-off-by: minmingzhu <minming.zhu@intel.com>
oap-project · Aug 9, 2023 · c199710 · c199710
1 parent fd0cd1d
commit c199710
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/OneDAL.scala
@@ -452,7 +452,7 @@ object OneDAL {
       val numCols = list(0).getAs[Vector](1).toArray.size
 
       val labelsArray = new Array[Double](numRows)
-      val featuresAddress= OneDAL.cNewFloatArray(numRows.toLong * numCols)
+      val featuresAddress= OneDAL.cNewDoubleArray(numRows.toLong * numCols)
       for ( i <- 0 until  numberCores) {
         val f = Future {
           val iter = list.iterator
@@ -626,7 +626,7 @@ object OneDAL {
       val numRows = list.size
       val numCols = list(0).toArray.size
       val size = numRows.toLong * numCols.toLong
-      val targetArrayAddress = OneDAL.cNewFloatArray(size)
+      val targetArrayAddress = OneDAL.cNewDoubleArray(size)
       for ( i <- 0 until  numberCores) {
         val f = Future {
           val iter = list.iterator