From f414239ed538a221af5286f75dbd883c939fe6a5 Mon Sep 17 00:00:00 2001 From: Alexander Stoyakin Date: Fri, 30 Aug 2019 16:18:41 +0300 Subject: [PATCH] Types fixed (#205) Signed-off-by: Alexander Stoyakin --- .../test/java/org/deeplearning4j/plot/BarnesHutTsneTest.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/plot/BarnesHutTsneTest.java b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/plot/BarnesHutTsneTest.java index aba60fe4f..f65783bce 100644 --- a/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/plot/BarnesHutTsneTest.java +++ b/deeplearning4j/deeplearning4j-core/src/test/java/org/deeplearning4j/plot/BarnesHutTsneTest.java @@ -498,9 +498,10 @@ public class BarnesHutTsneTest extends BaseDL4JTest { 0.08651136699915507, 0.7445210640026082, 0.6547649514127559, 0.3384719042666908, 0.05816723105860, 0.6248951423054205, 0.7431868493349041}; INDArray ndinput = Nd4j.createFromArray(input).reshape(11, 5); - double[] rows = {0, 10.0000, 20.0000, 30.0000, 40.0000, 50.0000, 60.0000, 69.0000, 78.0000, 88.0000, 98.0000, 108.0000}; + int[] rows = {0, 10, 20, 30, 40, 50, 60, 69, 78, 88, 98, 108}; INDArray indRows = Nd4j.createFromArray(rows); - double[] cols = {4.0000, 3.0000, 10.0000, 8.0000, 6.0000, 7.0000, 1.0000, 5.0000, 9.0000, 2.0000, 0, 4.0000, 9.0000, 8.0000, 10.0000, 2.0000, 6.0000, 7.0000, 3.0000, 5.0000, 1.0000, 6.0000, 8.0000, 3.0000, 9.0000, 10.0000, 4.0000, 0, 5.0000, 7.0000, 0, 1.0000, 2.0000, 10.0000, 4.0000, 6.0000, 8.0000, 9.0000, 5.0000, 7.0000, 0, 1.0000, 2.0000, 3.0000, 10.0000, 8.0000, 9.0000, 6.0000, 7.0000, 5.0000, 0, 2.0000, 3.0000, 7.0000, 9.0000, 10.0000, 4.0000, 8.0000, 1.0000, 6.0000, 0, 1.0000, 2.0000, 3.0000, 4.0000, 8.0000, 10.0000, 9.0000, 5.0000, 0, 1.0000, 3.0000, 4.0000, 5.0000, 9.0000, 10.0000, 8.0000, 2.0000, 0, 1.0000, 2.0000, 3.0000, 4.0000, 5.0000, 6.0000, 7.0000, 10.0000, 9.0000, 0, 1.0000, 2.0000, 3.0000, 4.0000, 5.0000, 6.0000, 7.0000, 8.0000, 10.0000, 0, 1.0000, 2.0000, 3.0000, 4.0000, 5.0000, 6.0000, 7.0000, 8.0000, 9.0000}; + int[] cols = {4, 3, 10, 8, 6, 7, 1, 5, 9, 2, 0, 4, 9, 8, 10, 2, 6, 7, 3, 5, 1, 6, 8, 3, 9, 10, 4, 0, 5, 7, 0, 1, 2, 10, 4, 6, 8, 9, + 5, 7, 0, 1, 2, 3, 10, 8, 9, 6, 7, 5, 0, 2, 3, 7, 9, 10, 4, 8, 1, 6, 0, 1, 2, 3, 4, 8, 10, 9, 5, 0, 1, 3, 4, 5, 9, 10, 8, 2, 0, 1, 2, 3, 4, 5, 6, 7, 10, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; INDArray indCols = Nd4j.createFromArray(cols); double[] vals = {0.6806, 0.1978, 0.1349, 0.0403, 0.0087, 0.0369, 0.0081, 0.0172, 0.0014, 0.0046, 0.0081, 0.3375, 0.2274, 0.0556, 0.0098, 0.0175, 0.0027, 0.0077, 0.0014, 0.0023, 0.0175, 0.6569, 0.1762, 0.0254, 0.0200, 0.0118, 0.0074, 0.0046, 0.0124, 0.0012, 0.1978, 0.0014, 0.0254, 0.7198, 0.0712, 0.0850, 0.0389, 0.0555, 0.0418, 0.0286, 0.6806, 0.3375, 0.0074, 0.0712, 0.2290, 0.0224, 0.0189, 0.0080, 0.0187, 0.0097, 0.0172, 0.0124, 0.0418, 0.7799, 0.0521, 0.0395, 0.0097, 0.0030, 0.0023, 1.706e-5, 0.0087, 0.0027, 0.6569, 0.0850, 0.0080, 0.5562, 0.0173, 0.0015, 1.706e-5, 0.0369, 0.0077, 0.0286, 0.0187, 0.7799, 0.0711, 0.0200, 0.0084, 0.0012, 0.0403, 0.0556, 0.1762, 0.0389, 0.0224, 0.0030, 0.5562, 0.0084, 0.0060, 0.0028, 0.0014, 0.2274, 0.0200, 0.0555, 0.0189, 0.0521, 0.0015, 0.0711, 0.0028, 0.3911, 0.1349, 0.0098, 0.0118, 0.7198, 0.2290, 0.0395, 0.0173, 0.0200, 0.0060, 0.3911}; INDArray indVals = Nd4j.createFromArray(vals);