Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -91,9 +91,9 @@ def run_models(task_type, dataset_name):
|
|
| 91 |
|
| 92 |
models = {
|
| 93 |
"Logistic Regression": LogisticRegression(max_iter=1000),
|
|
|
|
| 94 |
"Decision Tree": DecisionTreeClassifier(),
|
| 95 |
"Random Forest": RandomForestClassifier(),
|
| 96 |
-
"KNN": KNeighborsClassifier(),
|
| 97 |
}
|
| 98 |
|
| 99 |
# =========================
|
|
@@ -117,9 +117,9 @@ def run_models(task_type, dataset_name):
|
|
| 117 |
|
| 118 |
models = {
|
| 119 |
"Linear Regression": LinearRegression(),
|
|
|
|
| 120 |
"Decision Tree": DecisionTreeRegressor(),
|
| 121 |
"Random Forest": RandomForestRegressor(),
|
| 122 |
-
"SVR": SVR(),
|
| 123 |
}
|
| 124 |
|
| 125 |
# =========================
|
|
|
|
| 91 |
|
| 92 |
models = {
|
| 93 |
"Logistic Regression": LogisticRegression(max_iter=1000),
|
| 94 |
+
"SVM": SVC(),
|
| 95 |
"Decision Tree": DecisionTreeClassifier(),
|
| 96 |
"Random Forest": RandomForestClassifier(),
|
|
|
|
| 97 |
}
|
| 98 |
|
| 99 |
# =========================
|
|
|
|
| 117 |
|
| 118 |
models = {
|
| 119 |
"Linear Regression": LinearRegression(),
|
| 120 |
+
"SVR": SVR(),
|
| 121 |
"Decision Tree": DecisionTreeRegressor(),
|
| 122 |
"Random Forest": RandomForestRegressor(),
|
|
|
|
| 123 |
}
|
| 124 |
|
| 125 |
# =========================
|