Update backend/app/api/experiments.py

backend/app/api/experiments.py

@@ -24,11 +24,20 @@ def run_experiment(req: RunRequest, db: Session = Depends(get_db)):
     if not dataset:
         return {"error": "dataset not found"}
 
+    # load data
     df = pd.read_csv(dataset.file_path)
 
-    X = df[req.feature_columns]
+    # select features
+    X = df[req.feature_columns].copy()
 
-    model = KMeans(n_clusters=req.n_clusters)
+    # 🔥 FIX: encode categorical columns
+    X = pd.get_dummies(X)
+
+    # fill missing values
+    X = X.fillna(0)
+
+    # run clustering
+    model = KMeans(n_clusters=req.n_clusters, n_init=10)
     labels = model.fit_predict(X)
 
     return {