diff --git a/src/lib/ssbtoolkit.py b/src/lib/ssbtoolkit.py
index bb2e12f7ce52cd340cdb929fdf9c3f926b90b3c7..278326eef8f32b4a43242c18e840b55c3c969a8f 100644
--- a/src/lib/ssbtoolkit.py
+++ b/src/lib/ssbtoolkit.py
@@ -1559,10 +1559,11 @@ class simulation:
return
- def plotIterations(self):
+ def plotIterations(self, save=False, filename=None):
'''
Plots iterations
'''
+ import plotly.offline as pyoff
#dependencies
if self._iteration == None: raise TypeError('Simulation data not exist. simulation.fitModel.run() must be run first.')
@@ -1598,6 +1599,16 @@ class simulation:
)
fig = go.Figure(data=[trace], layout=layout)
+ if save==True:
+ if filename==None:
+ filename='plot.html'
+ return pyoff.plot(fig, filename=filename)
+ else:
+ ext = os.path.splitext(filename)[-1]
+ if ext == '.png': fig.write_image(filename, scale=3)
+ elif ext == '.html': pyoff.plot(fig, filename=filename)
+ else: raise TypeError("extension not valid. Use png or html.")
+ elif save ==False: return fig
return fig
def plotCurves(self, save=False, filename=None):