Skip to content

Commit 3642b01

Browse files
alexbarrosfabclmnt
authored andcommitted
feat: add adfuller stationary test parameters to config
1 parent fc97cc6 commit 3642b01

2 files changed

Lines changed: 9 additions & 4 deletions

File tree

src/ydata_profiling/config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""Configuration for the package."""
22
from enum import Enum
33
from pathlib import Path
4-
from typing import Any, Dict, List, Optional, Tuple, Union
4+
from typing import Any, Dict, List, Literal, Optional, Tuple, Union
55

66
import yaml
77
from pydantic.v1 import BaseModel, BaseSettings, Field, PrivateAttr
@@ -112,6 +112,8 @@ class TimeseriesVars(BaseModel):
112112
lags: List[int] = [1, 7, 12, 24, 30]
113113
significance: float = 0.05
114114
pacf_acf_lag: int = 100
115+
autolag: Optional[Literal["AIC", "BIC", "t-stat"]] = "AIC"
116+
maxlag: Optional[int] = None
115117

116118

117119
class Univariate(BaseModel):

src/ydata_profiling/model/pandas/describe_timeseries_pandas.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,15 @@
1616

1717

1818
def stationarity_test(config: Settings, series: pd.Series) -> Tuple[bool, float]:
19-
significance_threshold = config.vars.timeseries.significance
20-
2119
# make sure the data has no missing values
22-
adfuller_test = adfuller(series.dropna())
20+
adfuller_test = adfuller(
21+
series.dropna(),
22+
autolag=config.vars.timeseries.autolag,
23+
maxlag=config.vars.timeseries.maxlag
24+
)
2325
p_value = adfuller_test[1]
2426

27+
significance_threshold = config.vars.timeseries.significance
2528
return p_value < significance_threshold, p_value
2629

2730

0 commit comments

Comments
 (0)