diff --git a/causaltune/dataset_processor.py b/causaltune/dataset_processor.py index de06ee31..fb8deee4 100644 --- a/causaltune/dataset_processor.py +++ b/causaltune/dataset_processor.py @@ -10,7 +10,7 @@ class CausalityDatasetProcessor(BaseEstimator, TransformerMixin): """ A processor for CausalityDataset, designed to preprocess data for causal inference tasks by encoding, normalizing, - and handling missing values + and handling missing values. Attributes: encoder_type (str): Type of encoder used for categorical feature encoding ('onehot', 'label', 'target', 'woe'). outcome (str): The target variable used for encoding.