Create experiments.py

backend/app/api/experiments.py

@@ -0,0 +1,38 @@
+import pandas as pd
+from fastapi import APIRouter, Depends
+from pydantic import BaseModel
+from sqlalchemy.orm import Session
+from sklearn.cluster import KMeans
+
+from backend.app.db import get_db
+from backend.app.repositories.dataset_repo import get_dataset
+
+router = APIRouter(tags=["experiments"])
+
+
+class RunRequest(BaseModel):
+    dataset_id: str
+    algorithm: str = "kmeans"
+    n_clusters: int = 4
+    feature_columns: list[str]
+
+
+@router.post("/experiments/run")
+def run_experiment(req: RunRequest, db: Session = Depends(get_db)):
+    dataset = get_dataset(db, req.dataset_id)
+
+    if not dataset:
+        return {"error": "dataset not found"}
+
+    df = pd.read_csv(dataset.file_path)
+
+    X = df[req.feature_columns]
+
+    model = KMeans(n_clusters=req.n_clusters)
+    labels = model.fit_predict(X)
+
+    return {
+        "experiment_id": "exp_demo",
+        "status": "completed",
+        "clusters": labels.tolist(),
+    }