[This PR adds 5 clustering models corrected {sagnik1511#32}]

Affinity Propagation Agglomerative Clustering Birch DBSCAN KMeans
Tihsrah · Feb 11, 2022 · efa8117 · efa8117
1 parent 4c77bd9
commit efa8117
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py
@@ -17,7 +17,7 @@
         "Linear Regression": LinearRegression,
         "Lasso Regression": Lasso,
         "Ridge Regression": Ridge,
-        "Random Forest Regression": RandomForestRegressor,
+        "Random Forest Regression": RandomForestRegressor
     },
     "classification": {
         "Decision Tree Classifier": DecisionTreeClassifier,