From dde5c6ca68dcc471c2f75894a86da4f791e9dbaf Mon Sep 17 00:00:00 2001 From: Tihsrah Date: Sat, 12 Feb 2022 00:50:38 +0530 Subject: [PATCH 1/2] 5 clustering --- tab_automl/automl/models.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py index a84a6d2..da412e6 100644 --- a/tab_automl/automl/models.py +++ b/tab_automl/automl/models.py @@ -9,7 +9,11 @@ from sklearn.tree import DecisionTreeClassifier from sklearn.neighbors import KNeighborsClassifier from sklearn.neighbors import KNeighborsRegressor - +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": { @@ -26,5 +30,12 @@ "Random Forest Classifier": RandomForestClassifier, "XGBoost Classifier": XGBClassifier, "KNN Classifier": KNeighborsClassifier + }, + "clustering":{ + "Affinity Propagation":AffinityPropagation, + "Agglomerative Clustering":AgglomerativeClustering, + "Birch":Birch, + "DBSCAN":DBSCAN, + "KMeans":KMeans } } From fff70d2d2af46e330353769f51ff750679e0f458 Mon Sep 17 00:00:00 2001 From: Tihsrah Date: Sat, 12 Feb 2022 23:08:37 +0530 Subject: [PATCH 2/2] Added proper spacing between the keys and the values of the clustering dict. --- tab_automl/automl/models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py index da412e6..198585d 100644 --- a/tab_automl/automl/models.py +++ b/tab_automl/automl/models.py @@ -31,11 +31,11 @@ "XGBoost Classifier": XGBClassifier, "KNN Classifier": KNeighborsClassifier }, - "clustering":{ - "Affinity Propagation":AffinityPropagation, - "Agglomerative Clustering":AgglomerativeClustering, - "Birch":Birch, - "DBSCAN":DBSCAN, - "KMeans":KMeans + "clustering": { + "Affinity Propagation": AffinityPropagation, + "Agglomerative Clustering": AgglomerativeClustering, + "Birch": Birch, + "DBSCAN": DBSCAN, + "KMeans": KMeans } }