saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-01 03:45:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Changed returned object of plot_plotly, addressing #937 (#954)

Browse source
This commit is contained in:
Olof Höjvall 2019-05-06 22:55:25 +02:00 committed by Ben Letham
parent ec7689f03e
commit ee14933b3f
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
python/fbprophet/plot.py
View file

@ -27,7 +27,6 @@ except ImportError:
try:
import plotly.graph_objs as go
import plotly.offline
except ImportError:
logger.error('Importing plotly failed. Interactive plots will not work.')
@ -507,7 +506,12 @@ def plot_plotly(m, fcst, uncertainty=True, plot_cap=True, trend=False, changepoi
changepoints_threshold=0.01, xlabel='ds', ylabel='y'):
"""Plot the Prophet forecast with Plotly offline.
Requires plotly.offline.init_notebook_mode() to have been run,
Plotting in Jupyter Notebook requires initializing plotly.offline.init_notebook_mode():
>>> import plotly.offline as py
>>> py.init_notebook_mode()
Then the figure can be displayed using plotly.offline.iplot(...):
>>> fig = plot_plotly(m, fcst)
>>> py.iplot(fig)
see https://plot.ly/python/offline/ for details
Parameters
@ -525,7 +529,7 @@ def plot_plotly(m, fcst, uncertainty=True, plot_cap=True, trend=False, changepoi
Returns
-------
A Plotly plot.
A Plotly Figure.
"""
prediction_color = '#0072B2'
error_color = 'rgba(0, 114, 178, 0.2)' # '#0072B2' with 0.2 opacity
@ -649,4 +653,4 @@ def plot_plotly(m, fcst, uncertainty=True, plot_cap=True, trend=False, changepoi
),
)
fig = go.Figure(data=data, layout=layout)
return plotly.offline.iplot(fig)
return fig