Adding setter method for AVLTreeDigest gen object

tdunning · Jan 4, 2022 · c670bba · c670bba
1 parent addee2b
commit c670bba
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/core/src/main/java/com/tdunning/math/stats/AVLTreeDigest.java b/core/src/main/java/com/tdunning/math/stats/AVLTreeDigest.java
@@ -47,6 +47,17 @@ public AVLTreeDigest(double compression) {
         summary = new AVLGroupTree(false);
     }
 
+    /**
+     * Sets the seed for the RNG.
+     * In cases where a predicatable tree should be created, this function may be used to make the
+     * randomness in this AVLTree become more deterministic.
+     *
+     * @param seed The random seed to use for RNG purposes
+     */
+    public void setRandomSeed(long seed) {
+        gen.setSeed(seed);
+    }
+
     @Override
     public TDigest recordAllData() {
         if (summary.size() != 0) {