Fix n_neighbors

app.py

@@ -59,8 +59,19 @@ dataset_mapping = {
     "Fraud Detection": "datasets/fraud_detection.csv",
     "Customer Segmentation": "datasets/customer_segmentation.csv",
     "Loan Approval": "datasets/loan_approval.csv",
+    "Churn Prediction": "datasets/churn_prediction.csv",
+    "Handwritten Digit Recognition": "datasets/handwritten_digit_recognition.csv",
+    "Sentiment Analysis": "datasets/sentiment_analysis.csv",
     "House Price Prediction": "datasets/house_price_prediction.csv",
+    "Stock Prediction": "datasets/stock_prediction.csv",
     "Sales Forecasting": "datasets/sales_forecasting.csv",
+    "Stock Market Trends": "datasets/stock_market_trends.csv",
+    "Energy Consumption": "datasets/energy_consumption.csv",
+    "Patient Survival Prediction": "datasets/patient_survival_prediction.csv",
+    "House Price Estimation": "datasets/house_price_estimation.csv",
+    "Revenue Prediction": "datasets/revenue_prediction.csv",
+    "Weather Forecasting": "datasets/weather_forecasting.csv",
+    "Traffic Flow Prediction": "datasets/traffic_flow_prediction.csv"
 }
 
 dataset_path = dataset_mapping.get(problem, "datasets/spam_detection.csv")
@@ -69,7 +80,7 @@ df = pd.read_csv(dataset_path)
 
 # Model Initialization
 model_mapping = {
-    "KNN": KNeighborsClassifier() if task == "Classification" else KNeighborsRegressor(),
+    "KNN": KNeighborsClassifier(n_neighbors=min(5, len(y_train))) if task == "Classification" else KNeighborsRegressor(),
     "SVM": SVC() if task == "Classification" else SVR(),
     "Random Forest": RandomForestClassifier() if task == "Classification" else RandomForestRegressor(),
     "Decision Tree": DecisionTreeClassifier() if task == "Classification" else DecisionTreeRegressor(),