saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-09-16 22:20:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make plot axis label names configurable

Browse source
This commit is contained in:
Arne Sund 2017-02-26 14:38:10 +01:00
parent e08cfd2176
commit 8dd857d551
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
python/fbprophet/forecaster.py
View file

@ -618,13 +618,15 @@ class Prophet(object):
return pd.DataFrame({'ds': dates})
def plot(self, fcst, uncertainty=True):
def plot(self, fcst, uncertainty=True, xlabel='ds', ylabel='y'):
"""Plot the Prophet forecast.
Parameters
----------
fcst: pd.DataFrame output of self.predict.
uncertainty: Optional boolean to plot uncertainty intervals.
xlabel: Optional label name on X-axis
ylabel: Optional label name on Y-axis
Returns
-------
@ -641,8 +643,8 @@ class Prophet(object):
ax.fill_between(fcst['ds'].values, fcst['yhat_lower'],
fcst['yhat_upper'], color=forecast_color, alpha=0.2)
ax.grid(True, which='major', c='gray', ls='-', lw=1, alpha=0.2)
ax.set_xlabel('ds')
ax.set_ylabel('y')
ax.set_xlabel(xlabel)
ax.set_ylabel(ylabel)
fig.tight_layout()
return fig