2 files changed, 13 insertions, 0 deletions

diff --git a/Metrics/Metrics-Calculation/metrics_plot/utils/DistributionMetrics.py b/Metrics/Metrics-Calculation/metrics_plot/utils/DistributionMetrics.py
index 6e707108..cf532bc5 100644
--- a/Metrics/Metrics-Calculation/metrics_plot/utils/DistributionMetrics.py
+++ b/Metrics/Metrics-Calculation/metrics_plot/utils/DistributionMetrics.py
@@ -5,6 +5,16 @@ def ks_distance(samples1, samples2):
     value, p = stats.ks_2samp(samples1, samples2)
     return (value, p)
 
+def manual_ks(pdf1, pdf2):
+    result = 0
+    sum1 = 0
+    sum2 = 0
+    for(a, b) in zip(pdf1, pdf2):
+        sum1 += a
+        sum2 += b
+        result = max(result, abs(sum1-sum2))
+    return result
+
 def js_distance(samples1, samples2):
     map1 = fromSamples(samples1)
     map2 = fromSamples(samples2)
diff --git a/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py b/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py
index 0a9a95c5..41054294 100644
--- a/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py
+++ b/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py
@@ -10,6 +10,7 @@ class GraphCollection:
         self.out_ds = []
         self.nas = []
         self.mpcs = []
+        self.nts = []
         self.name = name
         models = reader.readmultiplefiles(path, number, shouldShuffle)
         print(len(models))
@@ -19,6 +20,8 @@ class GraphCollection:
             self.out_ds.append(out_d)
             self.nas.append(na)
             self.mpcs.append(mpc)
+            if(constants.Node_TYPE_KEY in contents):
+                self.nts.append(contents[constants.Node_TYPE_KEY])
 
 #Graph stat for one graph
 class GraphStat: