saja003 commited on
Commit
384fef3
·
verified ·
1 Parent(s): e476f7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
  # =========================