Skip to content

Commit

Permalink
Update dataset_processor.py
Browse files Browse the repository at this point in the history
Signed-off-by: AlxdrPolyakov <[email protected]>
  • Loading branch information
AlxdrPolyakov authored Sep 12, 2024
1 parent 6c8be18 commit 1210ff8
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions causaltune/dataset_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,16 @@


class CausalityDatasetProcessor(BaseEstimator, TransformerMixin):
"""
A custom dataset processor for CausalTune experiments, designed to preprocess datasets by encoding categorical features,
normalizing numerical features, and handling missing values for causal inference tasks.
Attributes:
encoder_type (str): The type of encoder used for categorical feature encoding (e.g., 'onehot', 'label', 'target', 'woe').
outcome (str): The target variable or outcome used for encoding and modeling.
encoder: The encoder object used during feature transformations.
"""

def __init__(self):
self.encoder_type = None
self.outcome = None
Expand Down

0 comments on commit 1210ff8

Please sign in to comment.