| 59 | return processed_data |
| 60 | |
| 61 | class FeatureEngineer: |
| 62 | |
| 63 | |
| 64 | def engineer_features(self, data: pd.DataFrame) -> pd.DataFrame: |
| 65 | |
| 66 | engineered_data = data.copy() |
| 67 | |
| 68 | # Add interaction terms between numerical features |
| 69 | numerical_cols = engineered_data.select_dtypes(include=['int64', 'float64']).columns |
| 70 | for i in range(len(numerical_cols)): |
| 71 | for j in range(i + 1, len(numerical_cols)): |
| 72 | col1, col2 = numerical_cols[i], numerical_cols[j] |
| 73 | engineered_data[f'{col1}_{col2}_interaction'] = ( |
| 74 | engineered_data[col1] * engineered_data[col2] |
| 75 | ) |
| 76 | |
| 77 | return engineered_data |
| 78 | |
| 79 | class ModelSelector: |
| 80 | |