Merge pull request #18 from snega16/snega16

[Issue {#7}] Add new models for classification training . Thanks @snega16 , happy contributing.🎉

requirements.txt

@@ -1,4 +1,5 @@
 lightgbm~=3.3.2
+xgboost~=1.5.2
 scikit-learn~=1.0.2
 pandas~=1.3.5
-numpy~=1.22.0
+numpy~=1.22.0

tab_automl/automl/models.py

@@ -2,6 +2,7 @@
 This files holds all codes for defined and used models through the life cycle of the package.
 """
 from lightgbm import LGBMClassifier
+from xgboost import XGBClassifier
 from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
 from sklearn.linear_model import Lasso, LinearRegression, Ridge
 from sklearn.tree import DecisionTreeClassifier
@@ -17,6 +18,7 @@
     "classification": {
         "Decision Tree Classifier": DecisionTreeClassifier,
         "Light Gradient Boosting Classifier": LGBMClassifier,
-        "Random Forest Classifier": RandomForestClassifier
+        "Random Forest Classifier": RandomForestClassifier,
+        "XGBoost Classifier": XGBClassifier
     }
 }