Update model/train.py

model/train.py

@@ -1150,7 +1150,14 @@ class EnhancedModelTrainer:
 
             # Save individual components for backward compatibility
             try:
-                if hasattr(model, 'named_steps'):
+                if model_name == 'ensemble':
+                    # Handle ensemble model saving - ensemble has different structure
+                    joblib.dump(model, self.model_path)
+                    logger.info(f"✅ Saved ensemble model to {self.model_path}")
+                    
+                    # Don't try to extract individual components from ensemble
+                    
+                elif hasattr(model, 'named_steps'):
                     if 'model' in model.named_steps:
                         joblib.dump(model.named_steps['model'], self.model_path)
                         logger.info(f"✅ Saved model component to {self.model_path}")
@@ -1168,10 +1175,6 @@ class EnhancedModelTrainer:
                         }
                         joblib.dump(enhanced_ref, self.vectorizer_path)
                         logger.info(f"✅ Saved enhanced features reference to {self.vectorizer_path}")
-                elif model_name == 'ensemble':
-                    # Handle ensemble model saving
-                    joblib.dump(model, self.model_path)
-                    logger.info(f"✅ Saved ensemble model to {self.model_path}")
                         
             except Exception as e:
                 logger.warning(f"Could not save individual components: {e}")