# HG changeset patch
# User Nicolas Saunier <nicolas.saunier@polymtl.ca>
# Date 1542216650 18000
# Node ID 0680387a89bbd4cffa3a2b7243e1c973735b51d3
# Parent  9cc51a2d3c4649e951bcd8853174bfc4e40905c2
added basic saving capability to empirical distribution

diff -r 9cc51a2d3c46 -r 0680387a89bb trafficintelligence/utils.py
--- a/trafficintelligence/utils.py	Wed Oct 31 16:25:10 2018 -0400
+++ b/trafficintelligence/utils.py	Wed Nov 14 12:30:50 2018 -0500
@@ -144,7 +144,19 @@
         super(EmpiricalContinuousDistribution, self).__init__(**kwargs)
         self.values = values
         self.probabilities = probabilities
-    
+
+    def save(self, filename):
+        import yaml
+        with open(filename, 'w') as out:
+            yaml.dump([self.values, self.probabilities], out)
+
+    @staticmethod
+    def load(filename):
+        import yaml
+        with open(filename) as f:
+            values, probabilities = yaml.load(f)
+            return EmpiricalContinuousDistribution(values, probabilities)
+
     def _cdf(self, x):
         if x < self.values[0]:
             return self.probabilities[0]