Add max score normalisation

online-ml · Oct 21, 2023 · ee78b8e · ee78b8e
1 parent 7ea3985
commit ee78b8e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/anomaly/half_space_tree.rs b/src/anomaly/half_space_tree.rs
@@ -192,6 +192,7 @@ impl<F: Float + FromPrimitive + AddAssign + SubAssign + MulAssign + DivAssign> H
             }
         }
         if do_score {
+            score = F::one() - (score / self.max_score());
             return Some(score);
         }
         return None;
@@ -202,4 +203,9 @@ impl<F: Float + FromPrimitive + AddAssign + SubAssign + MulAssign + DivAssign> H
     pub fn score_one(&mut self, observation: &Observation<F>) -> Option<F> {
         self.update(observation, true, false)
     }
+    fn max_score(&self) -> F {
+        F::from(self.n_trees).unwrap()
+            * F::from(self.window_size).unwrap()
+            * (F::from(2.).unwrap().powi(self.height as i32 + 1) - F::one())
+    }
 }