Added getProbabilityDistribution method.

StarlangSoftware · Apr 24, 2021 · 7a8044d · 7a8044d
1 parent fb539f4
commit 7a8044d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Sources/Math/DiscreteDistribution.swift b/Sources/Math/DiscreteDistribution.swift
@@ -215,6 +215,14 @@ public class DiscreteDistribution{
             return 0.0
         }
     }
+
+    public func getProbabilityDistribution() -> [String : Double]{
+        var result : [String: Double] = [:]
+        for item in keys{
+            result[item] = getProbability(item: item)
+        }
+        return result
+    }
 
     /**
     The getProbabilityLaplaceSmoothing method takes an item as an input returns the smoothed value to which the