Spaces:
Sleeping
Sleeping
Surbhi
commited on
Commit
·
f03eea4
1
Parent(s):
ccf2ebc
Fix model
Browse files
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=
|
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(),
|