diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py
index 4502e4b..6dc6a79 100644
--- 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
+    }
     
 }