Surbhi commited on
Commit
f03eea4
·
1 Parent(s): ccf2ebc
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ df = pd.read_csv(dataset_path)
80
 
81
  # Model Initialization
82
  model_mapping = {
83
- "KNN": KNeighborsClassifier(n_neighbors=min(5, len(y_train))) if task == "Classification" else KNeighborsRegressor(),
84
  "SVM": SVC() if task == "Classification" else SVR(),
85
  "Random Forest": RandomForestClassifier() if task == "Classification" else RandomForestRegressor(),
86
  "Decision Tree": DecisionTreeClassifier() if task == "Classification" else DecisionTreeRegressor(),
 
80
 
81
  # Model Initialization
82
  model_mapping = {
83
+ "KNN": KNeighborsClassifier(n_neighbors=5) if task == "Classification" else KNeighborsRegressor(),
84
  "SVM": SVC() if task == "Classification" else SVR(),
85
  "Random Forest": RandomForestClassifier() if task == "Classification" else RandomForestRegressor(),
86
  "Decision Tree": DecisionTreeClassifier() if task == "Classification" else DecisionTreeRegressor(),