From db8fbbf0689b3d053e414aca79013bc17663046a Mon Sep 17 00:00:00 2001 From: ltcptgeneral <35508619+ltcptgeneral@users.noreply.github.com> Date: Tue, 5 May 2020 22:37:32 -0500 Subject: [PATCH] visualization.py v 1.0.0.001 --- analysis-master/analysis/visualization.py | 26 ++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) diff --git a/analysis-master/analysis/visualization.py b/analysis-master/analysis/visualization.py index 0d52c0f5..231a3a05 100644 --- a/analysis-master/analysis/visualization.py +++ b/analysis-master/analysis/visualization.py @@ -6,10 +6,12 @@ # fancy # setup: -__version__ = "1.0.0.000" +__version__ = "1.0.0.001" #changelog should be viewed using print(analysis.__changelog__) __changelog__ = """changelog: + 1.0.0.001: + - added graphhistogram function as a fragment of visualize_pit.py 1.0.0.000: - created visualization.py - added graphloss() @@ -26,9 +28,31 @@ __all__ = [ ] import matplotlib.pyplot as plt +import numpy as np def graphloss(losses): x = range(0, len(losses)) plt.plot(x, losses) + plt.show() + +def graphhistogram(data, figsize, sharey = True): # expects library with key as variable and contents as occurances + + fig, ax = plt.subplots(1, len(data), sharey=sharey, figsize=figsize) + + i = 0 + + for variable in data: + + ax[i].hist(data[variable]) + ax[i].invert_xaxis() + + ax[i].set_xlabel('Variable') + ax[i].set_ylabel('Frequency') + ax[i].set_title(variable) + + plt.yticks(np.arange(len(data[variable]))) + + i+=1 + plt.show() \ No newline at end of file