saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-30 20:18:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test for disabling uncertainty

Browse source
This commit is contained in:
Ryan Nazareth 2019-11-02 20:45:19 +00:00 committed by Ben Letham
parent ca9a49d328
commit c9e3a83188
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
python/fbprophet/tests/test_prophet.py
View file

@ -93,6 +93,19 @@ class TestProphet(TestCase):
fcst = m.predict(future)
self.assertEqual(fcst['yhat'].values[-1], 0)
def test_fit_predict_uncertainty_disabled(self):
N = DATA.shape[0]
train = DATA.head(N // 2)
future = DATA.tail(N // 2)
m = Prophet(uncertainty_samples=0)
m.fit(train)
fcst = m.predict(future)
self.assertNotIn('yhat_lower', list(fcst.columns))
self.assertNotIn('yhat_upper', list(fcst.columns))
self.assertNotIn('trend_lower', list(fcst.columns))
self.assertNotIn('trend_upper', list(fcst.columns))
def test_setup_dataframe(self):
m = Prophet()
N = DATA.shape[0]