From 799ff4846e3c9c9417c01be9671050424981088a Mon Sep 17 00:00:00 2001 From: Jonathan Shook Date: Fri, 8 Dec 2023 11:57:55 -0600 Subject: [PATCH] added scale to v_wrap --- .../vectors/dnn/DNN_euclidean_v_wrap.java | 22 ++++++++++++------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNN_euclidean_v_wrap.java b/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNN_euclidean_v_wrap.java index 7fe670ed1..c52b69297 100644 --- a/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNN_euclidean_v_wrap.java +++ b/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNN_euclidean_v_wrap.java @@ -26,20 +26,26 @@ import java.util.function.LongFunction; @Categories(Category.experimental) public class DNN_euclidean_v_wrap implements LongFunction { - private final int dimensions; - private final long population; + private final int D; + private final long N; + private final double scale; - public DNN_euclidean_v_wrap(int dimensions, long population) { - this.dimensions = dimensions; - this.population = population; + public DNN_euclidean_v_wrap(int D, long N, double scale) { + this.D = D; + this.N = N; + this.scale = scale; + } + + public DNN_euclidean_v_wrap(int D, long N) { + this(D,N,1.0d); } @Override public float[] apply(long value) { - value = value % population; - float[] vector = new float[dimensions]; + value = value % N; + float[] vector = new float[D]; for (int idx = 0; idx < vector.length; idx++) { - vector[idx]= (float)idx+value; + vector[idx]= (float)((idx+value)*scale); } return vector; }