Merge pull request sagnik1511#41 from PalaVenkiReddy/PalaVenkiReddy

#Issue-6  Added new models for Regression Training

tab_automl/automl/models.py

@@ -1,7 +1,7 @@
 """
 This files holds all codes for defined and used models through the life cycle of the package.
 """
-from lightgbm import LGBMClassifier
+from lightgbm import LGBMClassifier, LGBMRegressor
 from xgboost import XGBClassifier
 from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
 from sklearn.linear_model import Lasso, LinearRegression, Ridge
@@ -19,6 +19,7 @@
 from sklearn.cluster import OPTICS
 from sklearn.cluster import SpectralClustering
 from sklearn.mixture import GaussianMixture
+from sklearn.tree import DecisionTreeRegressor
 
 single_model_dict = {
     "regression": {
@@ -27,7 +28,9 @@
         "Ridge Regression": Ridge,
         "Random Forest Regression": RandomForestRegressor,
         "Support Vector Regression": SVR,
-        "KNN Regressor": KNeighborsRegressor
+        "KNN Regressor": KNeighborsRegressor,
+        "Decision Tree Regressor": DecisionTreeRegressor,
+        "Light Gradient Boosting Regressor": LGBMRegressor,
     },
     "classification": {
         "Decision Tree Classifier": DecisionTreeClassifier,