diff --git a/cobra/preprocessing/preprocessor.py b/cobra/preprocessing/preprocessor.py index 3338ec4..9ecfa50 100644 --- a/cobra/preprocessing/preprocessor.py +++ b/cobra/preprocessing/preprocessor.py @@ -172,8 +172,7 @@ class encapsulating CategoricalDataProcessor, target_encoder = TargetEncoder(weight, imputation_strategy) - return cls(model_type, - categorical_data_processor, discretizer, target_encoder) + return cls(categorical_data_processor, discretizer, target_encoder) @classmethod def from_pipeline(cls, pipeline: dict): @@ -211,8 +210,7 @@ def from_pipeline(cls, pipeline: dict): target_encoder = TargetEncoder() target_encoder.set_attributes_from_dict(pipeline["target_encoder"]) - return cls(model_type, - categorical_data_processor, discretizer, target_encoder, + return cls(categorical_data_processor, discretizer, target_encoder, is_fitted=pipeline["_is_fitted"]) def fit(self, train_data: pd.DataFrame, continuous_vars: list,