ref:Cleaned the Structure

ai_governance/data_processor.py

@@ -21,6 +21,7 @@ class DataProcessor:
         self.categorical_features = []
         self.feature_names = []
         self.encoders = {}
+        self.target_encoder = None  # Add target encoder
         self.scaler = StandardScaler()
         
         self.X_train = None
@@ -75,6 +76,13 @@ class DataProcessor:
         X = self.df[feature_cols].copy()
         y = self.df[self.target_column].copy()
         
+        # Encode target variable if it's categorical
+        if y.dtype == 'object' or y.dtype.name == 'category':
+            self.target_encoder = LabelEncoder()
+            y_encoded = self.target_encoder.fit_transform(y)
+            y = pd.Series(y_encoded, index=y.index)
+            print(f"Target '{self.target_column}' encoded: {dict(enumerate(self.target_encoder.classes_))}")
+        
         # Encode categorical variables
         for col in self.categorical_features:
             if col in X.columns: