pydtn agkmeans and version 1.0

Fatemeh's agkmeans, basic gui, prophet Version used for paper

examples/example2.py

@@ -7,6 +7,7 @@ from collections import namedtuple
 from os import path
 from multiprocessing import Pool
 from typing import ChainMap
+from numpy import mean
 
 import yaml
 import sys
@@ -121,7 +122,7 @@ def main(args):
     with open("testing.csv", "w", newline="") as results_file:
         for node_type in results:
             fieldnames = results[node_type][0].keys()
-            writer = csv.DictWriter(results_file, fieldnames=fieldnames)
+            writer = csv.DictWriter(results_file, fieldnames=fieldnames, extrasaction='ignore')  # extrasaction avoids intermittent dictwriter valueerror
             writer.writeheader()
             for result in results[node_type]:
                 writer.writerow(result)
@@ -136,11 +137,21 @@ def main(args):
 
         if len(dependant) > 1:      # avoids axes is not subscriptable error when plotting one graph
             for ax, dep in zip(axes, dependant):
-                y = [stat[dep] for stat in sorted_results]
+                yyys = [ [] for _ in range(len(input))]
+                # split the stats into batches
+                for stat in sorted_results:
+                    yyys[stat['tag']].append([stat[dep]])
+
+                y = [mean(stat) for stat in yyys]
                 ax.plot(x, y, "o-", label=node)
                 ax.set_ylabel(dep)
         else:
-            y = [stat[dependant[0]] for stat in sorted_results]
+            yyys = [ [] for _ in range(len(input))]
+            # split the stats into batches
+            for stat in sorted_results:
+                yyys[stat['tag']].append([stat[dependant[0]]])
+
+            y = [mean(stat) for stat in yyys]
             axes.plot(x, y, "o-", label=node)
             axes.set_ylabel(dependant[0])