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

Affinity Propagation Agglomerative Clustering Birch DBSCAN KMeans
Tihsrah · Feb 11, 2022 · 4c77bd9 · 4c77bd9
1 parent 84e66d3
commit 4c77bd9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py
@@ -25,8 +25,8 @@
         "Random Forest Classifier": RandomForestClassifier
     },
     "clustering":{
-        "AffinityPropagation":AffinityPropagation,
-        "AgglomerativeClustering":AgglomerativeClustering,
+        "Affinity Propagation":AffinityPropagation,
+        "Agglomerative Clustering":AgglomerativeClustering,
         "Birch":Birch,
         "DBSCAN":DBSCAN,
         "KMeans":KMeans