IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

adjusted scale factors for DNN vectors

Browse Source
This commit is contained in:
Jonathan Shook 2023-12-08 12:41:15 -06:00
parent 799ff4846e
commit 852f61ef61
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNN_euclidean_v.java
View File

@ -47,7 +47,7 @@ public class DNN_euclidean_v implements LongFunction<float[]> {
}
float[] vector = new float[D];
for (int idx = 0; idx < vector.length; idx++) {
vector[idx]= (float)((idx+value)*scale);
vector[idx]= (float)(value+(idx*scale));
}
return vector;
}

2
virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNN_euclidean_v_wrap.java
View File

@ -45,7 +45,7 @@ public class DNN_euclidean_v_wrap implements LongFunction<float[]> {
value = value % N;
float[] vector = new float[D];
for (int idx = 0; idx < vector.length; idx++) {
vector[idx]= (float)((idx+value)*scale);
vector[idx]= (float)(value+(idx*scale));
}
return vector;
}

2
virtdata-lib-basics/src/test/java/io/nosqlbench/virtdata/library/basics/shared/vectors/dnn/DNNEuclideanVTest.java
View File

@ -33,7 +33,7 @@ class DNNEuclideanVTest {
@Test
public void testBasicVectorsScaled() {
DNN_euclidean_v vf = new DNN_euclidean_v(5, 7, 3.0);
assertThat(vf.apply(3L)).isEqualTo(new float[]{9f,12f,15f,18f,21f});
assertThat(vf.apply(3L)).isEqualTo(new float[]{3f,6f,9f,12f,15f});
assertThrows(RuntimeException.class, () -> vf.apply(7));
}