add distance sum

1 files changed, 3 insertions, 1 deletions

diff --git a/km.h b/km.h
index 097026e..bc2e3a6 100644
--- a/km.h
+++ b/km.h
@@ -89,6 +89,7 @@ typedef struct {
 } km_row_map_t;
 
 typedef struct {
+  const float sum;
   const float *means;
   const float *variances;
   const size_t num_clusters;
@@ -112,7 +113,8 @@ km_solve(
 typedef struct {
   const km_set_t * const cluster_set;
 
-  const float mean_distance,
+  const float distance_sum,
+              mean_distance,
               mean_variance,
               mean_cluster_size;