From a882ad00515e730bad5e52fa29b74f461a5b9cd6 Mon Sep 17 00:00:00 2001
From: 20001LastOrder <boqi.chen@mail.mcgill.ca>
Date: Tue, 13 Aug 2019 18:10:02 -0400
Subject: change exploration value function

---
 Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py')

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:
-- 
cgit v1.2.3-70-g09d2