VectorDescriptorAggregator uses Double instead of Float

vitrivr · Aug 23, 2024 · a2ca74a · a2ca74a
1 parent 7fd87b5
commit a2ca74a
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/...main/kotlin/org/vitrivr/engine/index/aggregators/descriptor/VectorDescriptorAggregator.kt b/...main/kotlin/org/vitrivr/engine/index/aggregators/descriptor/VectorDescriptorAggregator.kt
@@ -26,50 +26,50 @@ class VectorDescriptorAggregator : TransformerFactory {
         MEAN {
             override fun aggregate(collection: Collection<VectorDescriptor<*, *>>): VectorDescriptor<*, *> {
 
-                val vec = FloatArray(collection.first().vector.size)
+                val vec = DoubleArray(collection.first().vector.size)
 
                 collection.forEach { descriptor ->
                     when (descriptor) {
                         is FloatVectorDescriptor -> {
                             for (i in vec.indices) {
-                                vec[i] += descriptor.vector.value[i]
+                                vec[i] += descriptor.vector.value[i].toDouble()
                             }
                         }
 
                         is DoubleVectorDescriptor -> {
                             for (i in vec.indices) {
-                                vec[i] += descriptor.vector.value[i].toFloat()
+                                vec[i] += descriptor.vector.value[i]
                             }
                         }
 
                         is IntVectorDescriptor -> {
                             for (i in vec.indices) {
-                                vec[i] += descriptor.vector.value[i].toFloat()
+                                vec[i] += descriptor.vector.value[i].toDouble()
                             }
                         }
 
                         is LongVectorDescriptor -> {
                             for (i in vec.indices) {
-                                vec[i] += descriptor.vector.value[i].toFloat()
+                                vec[i] += descriptor.vector.value[i].toDouble()
                             }
                         }
 
                         is BooleanVectorDescriptor -> {
                             for (i in vec.indices) {
-                                vec[i] += if (descriptor.vector.value[i]) 1f else 0f
+                                vec[i] += if (descriptor.vector.value[i]) 1.0 else 0.0
                             }
                         }
                     }
                 }
 
-                val div = collection.size.toFloat()
+                val div = collection.size.toDouble()
 
                 for (i in vec.indices) {
                     vec[i] /= div
                 }
 
-                return FloatVectorDescriptor(
-                    UUID.randomUUID(), null, Value.FloatVector(vec)
+                return DoubleVectorDescriptor(
+                    UUID.randomUUID(), null, Value.DoubleVector(vec)
                 )
             }
         };