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

Affinity Propagation Agglomerative Clustering Birch DBSCAN KMeans Push.
Tihsrah · Feb 11, 2022 · 584e9ae · 584e9ae
1 parent 66e256b
commit 584e9ae
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py
@@ -6,6 +6,11 @@
 from sklearn.linear_model import Lasso, LinearRegression, Ridge
 from sklearn.tree import DecisionTreeClassifier
 
+from sklearn.cluster import AffinityPropagation
+from sklearn.cluster import AgglomerativeClustering
+from sklearn.cluster import Birch
+from sklearn.cluster import DBSCAN
+from sklearn.cluster import KMeans
 
 single_model_dict = {
     "regression": {
@@ -19,5 +24,12 @@
         "Light Gradient Boosting Classifier": LGBMClassifier,
         "Random Forest Classifier": RandomForestClassifier
     },
+    "clustering":{
+        "Affinity Propagation":AffinityPropagation,
+        "Agglomerative Clustering":AgglomerativeClustering,
+        "Birch":Birch,
+        "DBSCAN":DBSCAN,
+        "KMeans":KMeans
+    }
 
 }