Use mse to avoid extra smooth in GSOM

reinterpretcat · Jan 25, 2025 · 5f694a7 · 5f694a7
1 parent d86b1ad
commit 5f694a7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/rosomaxa/src/algorithms/gsom/network.rs b/rosomaxa/src/algorithms/gsom/network.rs
@@ -483,7 +483,7 @@ where
     }
 
     /// Returns data (weights) dimension.
-    pub(super) fn dimension(&self) -> usize {
+    pub fn dimension(&self) -> usize {
         self.dimension
     }
 

diff --git a/rosomaxa/src/population/rosomaxa.rs b/rosomaxa/src/population/rosomaxa.rs
@@ -310,7 +310,12 @@ where
         network.set_learning_rate(get_learning_rate(statistics.termination_estimate));
 
         if statistics.generation % config.rebalance_memory == 0 {
-            network.smooth(external_ctx, 1, |i| i.on_update(external_ctx));
+            // set the MSE threshold to a fraction of the maximum possible normalized distance
+            let mse = network.mse();
+            let threshold = 0.5 / (network.dimension() as Float).sqrt();
+            if mse > threshold {
+                network.smooth(external_ctx, 1, |i| i.on_update(external_ctx));
+            }
         }
 
         let keep_size = get_keep_size(config.rebalance_memory, statistics.termination_estimate);