Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding clustering 5 models inside single_model_dict #32 #36

Merged
merged 2 commits into from
Feb 12, 2022
Merged

Adding clustering 5 models inside single_model_dict #32 #36

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
dde5c6c
5 clustering
Tihsrah Feb 11, 2022
fff70d2
Added proper spacing between the keys and the values of the clusterin…
Tihsrah Feb 12, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 12 additions & 1 deletion tab_automl/automl/models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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": {
Expand All @@ -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
}
}