CosineDistance extends FastDistance { @Override public double calc(double[] array1, double[] array2) { double dot = BLAS.dot(array1, array2); double cross = Math.sqrt(BLAS.dot(array1, array1) * BLAS.dot(array2, array2)); return 1.0 - (cross > 0.0 ? dot / cross : 0.0); }  @Override double calc(double[] array1, double[] array2); @Override double calc(Vector vec1, Vector vec2); @Override void updateLabel(FastDistanceData data);  }
@Test public void testContinuousDistance() { Assert.assertEquals(distance.calc(denseVector1, denseVector2), 0.2852, 0.01); Assert.assertEquals(distance.calc(denseVector1.getData(), denseVector2.getData()), 0.2852, 0.01); Assert.assertEquals(distance.calc(denseVector1, sparseVector1), 0.73, 0.01); Assert.assertEquals(distance.calc(sparseVector1, sparseVector2), 0.76, 0.01); Assert.assertEquals(distance.calc(sparseVector1, denseVector2), 0.60, 0.01); }
@Test public void testCalDistanceVecVec() { DenseMatrix denseResult = distance.calc(distance.prepareVectorData(Tuple2.of(denseVector1, null)), (FastDistanceData)distance.prepareVectorData(Tuple2.of(denseVector2, null))); Assert.assertEquals(denseResult.get(0, 0), 0.2852, 0.01); DenseMatrix sparseResult = distance.calc(distance.prepareVectorData(Tuple2.of(sparseVector1, null)), (FastDistanceData)distance.prepareVectorData(Tuple2.of(sparseVector2, null))); Assert.assertEquals(sparseResult.get(0, 0), 0.76, 0.01); }
@Test public void testCalDistanceMatrixMatrix() { FastDistanceMatrixData matrixData = initMatrixData(); DenseMatrix res = distance.calc(matrixData, matrixData); double[] expect = new double[] {0.0, 0.2852, 0.2852, 0.0}; double[] predict = res.getData(); for (int i = 0; i < expect.length; i++) { Assert.assertEquals(expect[i], predict[i], 0.01); } }
@Test public void testCalDistanceVecMatrix() { FastDistanceMatrixData matrixData = initMatrixData(); FastDistanceVectorData vectorData = distance.prepareVectorData(Row.of(0, sparseVector1), 1); double[] predict = distance.calc(matrixData, vectorData).getData(); double[] expect = new double[] {0.73, 0.60}; for (int i = 0; i < expect.length; i++) { Assert.assertEquals(expect[i], predict[i], 0.01); } predict = distance.calc(vectorData, matrixData).getData(); for (int i = 0; i < expect.length; i++) { Assert.assertEquals(expect[i], predict[i], 0.01); } }