Merge pull request #1032 from buttogo/code_interpreter

fix bug of old version param
This commit is contained in:
garylin2099 2024-03-19 13:49:41 +08:00 committed by GitHub
commit ce8ba61482
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -171,7 +171,7 @@ class OneHotEncode(DataPreprocessTool):
def __init__(self, features: list):
self.features = features
self.model = OneHotEncoder(handle_unknown="ignore", sparse=False)
self.model = OneHotEncoder(handle_unknown="ignore", sparse_output=False)
def transform(self, df: pd.DataFrame) -> pd.DataFrame:
ts_data = self.model.transform(df[self.features])