From 68af73f89c4df149e694c59471ae7761d78b9ff2 Mon Sep 17 00:00:00 2001 From: PalaVenkiReddy Date: Fri, 17 Feb 2023 18:04:01 +0530 Subject: [PATCH] Added new models for Classification Training --- tab_automl/automl/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py index fc65bd2..81d0fcb 100644 --- a/tab_automl/automl/models.py +++ b/tab_automl/automl/models.py @@ -20,6 +20,9 @@ from sklearn.cluster import SpectralClustering from sklearn.mixture import GaussianMixture from sklearn.tree import DecisionTreeRegressor +from sklearn.naive_bayes import GaussianNB +from sklearn.svm import LinearSVC +from sklearn.svm import SVC single_model_dict = { "regression": { @@ -37,7 +40,10 @@ "Light Gradient Boosting Classifier": LGBMClassifier, "Random Forest Classifier": RandomForestClassifier, "XGBoost Classifier": XGBClassifier, - "KNN Classifier": KNeighborsClassifier + "KNN Classifier": KNeighborsClassifier, + "Gaussian Naive Bayes": GaussianNB, + "Linear Support Vector Classifier": LinearSVC, + "Support Vector Classifier": SVC, }, "clustering": { "Affinity Propagation": AffinityPropagation,