fix(Config): rename sliding_window_size to initial_window_size (#235)

This commit is contained in:
Mark Aron Szulyovszky
2022-03-13 16:05:16 +01:00
committed by GitHub
parent 717e0ac979
commit 95b0499430
11 changed files with 19 additions and 19 deletions
+3 -3
View File
@@ -10,15 +10,15 @@ class PCATransformation(Transformation):
pca: PCA
def __init__(self, ratio_components_to_keep: float, sliding_window_size: int):
def __init__(self, ratio_components_to_keep: float, initial_window_size: int):
self.ratio_components_to_keep = ratio_components_to_keep
self.sliding_window_size = sliding_window_size
self.initial_window_size = initial_window_size
def fit(self, X: pd.DataFrame, y: Optional[pd.Series] = None) -> None:
self.pca = PCA(
n_components=min(
int(len(X.columns) * self.ratio_components_to_keep),
self.sliding_window_size,
self.initial_window_size,
)
)
self.pca.fit(X, y)