QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-18 23:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(prophet): enable confidence intervals and y_hat without forecast (#17658)

Browse Source 
* enable confidence intervals and y_hat without forecast

* fix if statement

Co-authored-by: Ville Brofeldt <33317356+villebro@users.noreply.github.com>

Co-authored-by: Ville Brofeldt <33317356+villebro@users.noreply.github.com>
This commit is contained in:
Shiva Raisinghani
2021-12-07 23:56:18 -08:00
committed by GitHub
parent 418c0b4e48
commit cd88b8e81e
3 changed files with 26 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/integration_tests/pandas_postprocessing_tests.py
View File

@@ -830,6 +830,28 @@ class TestPostProcessing(SupersetTestCase):
assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 5, 31)
assert len(df) == 9
def test_prophet_valid_zero_periods(self):
pytest.importorskip("prophet")
df = proc.prophet(
df=prophet_df, time_grain="P1M", periods=0, confidence_interval=0.9
)
columns = {column for column in df.columns}
assert columns == {
DTTM_ALIAS,
"a__yhat",
"a__yhat_upper",
"a__yhat_lower",
"a",
"b__yhat",
"b__yhat_upper",
"b__yhat_lower",
"b",
}
assert df[DTTM_ALIAS].iloc[0].to_pydatetime() == datetime(2018, 12, 31)
assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2021, 12, 31)
assert len(df) == 4
def test_prophet_import(self):
prophet = find_spec("prophet")
if prophet is None:
@@ -875,7 +897,7 @@ class TestPostProcessing(SupersetTestCase):
proc.prophet,
df=prophet_df,
time_grain="P1M",
periods=0,
periods=-1,
confidence_interval=0.8,
)