From 1210ff8e07bb64c0df6e24ef59e51e7ec3aa5b28 Mon Sep 17 00:00:00 2001
From: AlxdrPolyakov <122611538+AlxdrPolyakov@users.noreply.github.com>
Date: Thu, 12 Sep 2024 11:38:40 +0100
Subject: [PATCH] Update dataset_processor.py

Signed-off-by: AlxdrPolyakov <122611538+AlxdrPolyakov@users.noreply.github.com>
---
 causaltune/dataset_processor.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/causaltune/dataset_processor.py b/causaltune/dataset_processor.py
index 11ef5313..7941abff 100644
--- a/causaltune/dataset_processor.py
+++ b/causaltune/dataset_processor.py
@@ -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