Skip to content
Snippets Groups Projects

pydtn agkmeans and version 1.0

Merged Hunter McConnell (rtm534) requested to merge summer2022 into develop
1 file
+ 14
3
Compare changes
  • Side-by-side
  • Inline
+ 14
3
@@ -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])
Loading