Upload 4 files

trainer.py

@@ -36,9 +36,9 @@ def prepare_data(df: pd.DataFrame, target_col: str):
     # Detectar tipo de tarefa
     task_type = detect_task_type(y)
     
-    # Encode target se for classificação com strings
+    # Encode target - sempre para classificação (XGBoost precisa de 0, 1, 2...)
     label_encoder = None
-    if y.dtype == 'object':
+    if task_type == "classification":
         label_encoder = LabelEncoder()
         y = pd.Series(label_encoder.fit_transform(y))