unifiedbooster 0.2.2__py3-none-any.whl → 0.3.0__py3-none-any.whl

unifiedbooster/gbdt.py

@@ -7,6 +7,10 @@ class GBDT(BaseEstimator):
 
     Attributes:
 
+        model_type: str
+            type of gradient boosting algorithm: 'xgboost', 'lightgbm',
+            'catboost', 'gradientboosting'
+
         n_estimators: int
             maximum number of trees that can be built
 
@@ -24,6 +28,9 @@ class GBDT(BaseEstimator):
 
         seed: int
             reproducibility seed
+
+        **kwargs: dict
+            additional parameters to be passed to the class
     """
 
     def __init__(
@@ -83,6 +90,17 @@ class GBDT(BaseEstimator):
                 "bootstrap_type": "Bernoulli",
                 **kwargs,
             }
+        elif self.model_type == "gradientboosting":
+            self.params = {
+                "n_estimators": self.n_estimators,
+                "learning_rate": self.learning_rate,
+                "subsample": self.rowsample,
+                "max_features": self.colsample,
+                "max_depth": self.max_depth,
+                "verbose": self.verbose,
+                "random_state": self.seed,
+                **kwargs,
+            }
 
     def fit(self, X, y, **kwargs):
         """Fit custom model to training data (X, y).
@@ -109,6 +127,8 @@ class GBDT(BaseEstimator):
             self.n_classes_ = len(
                 self.classes_
             )  # for compatibility with sklearn
+        if getattr(self, "model_type") == "gradientboosting":
+            self.model.max_features = int(self.model.max_features * X.shape[1])
         return getattr(self, "model").fit(X, y, **kwargs)
 
     def predict(self, X):

unifiedbooster/gbdt_classification.py

@@ -7,6 +7,7 @@ try:
 except:
     print("catboost package can't be built")
 from lightgbm import LGBMClassifier
+from sklearn.ensemble import GradientBoostingClassifier
 
 
 class GBDTClassifier(GBDT, ClassifierMixin):
@@ -14,6 +15,10 @@ class GBDTClassifier(GBDT, ClassifierMixin):
 
     Attributes:
 
+        model_type: str
+            type of gradient boosting algorithm: 'xgboost', 'lightgbm',
+            'catboost', 'gradientboosting'
+
         n_estimators: int
             maximum number of trees that can be built
 
@@ -32,6 +37,9 @@ class GBDTClassifier(GBDT, ClassifierMixin):
         seed: int
             reproducibility seed
 
+        **kwargs: dict
+            additional parameters to be passed to the class
+
     Examples:
 
     ```python
@@ -105,6 +113,8 @@ class GBDTClassifier(GBDT, ClassifierMixin):
             self.model = CatBoostClassifier(**self.params)
         elif model_type == "lightgbm":
             self.model = LGBMClassifier(**self.params)
+        elif model_type == "gradientboosting":
+            self.model = GradientBoostingClassifier(**self.params)
         else:
             raise ValueError(f"Unknown model_type: {model_type}")
 

unifiedbooster/gbdt_regression.py

@@ -7,6 +7,7 @@ try:
 except:
     print("catboost package can't be built")
 from lightgbm import LGBMRegressor
+from sklearn.ensemble import GradientBoostingRegressor
 
 
 class GBDTRegressor(GBDT, RegressorMixin):
@@ -14,6 +15,10 @@ class GBDTRegressor(GBDT, RegressorMixin):
 
     Attributes:
 
+        model_type: str
+            type of gradient boosting algorithm: 'xgboost', 'lightgbm',
+            'catboost', 'gradientboosting'
+
         n_estimators: int
             maximum number of trees that can be built
 
@@ -32,6 +37,9 @@ class GBDTRegressor(GBDT, RegressorMixin):
         seed: int
             reproducibility seed
 
+        **kwargs: dict
+            additional parameters to be passed to the class
+
     Examples:
 
     ```python
@@ -105,5 +113,7 @@ class GBDTRegressor(GBDT, RegressorMixin):
             self.model = CatBoostRegressor(**self.params)
         elif model_type == "lightgbm":
             self.model = LGBMRegressor(**self.params)
+        elif model_type == "gradientboosting":
+            self.model = GradientBoostingRegressor(**self.params)
         else:
             raise ValueError(f"Unknown model_type: {model_type}")

{unifiedbooster-0.2.2.dist-info → unifiedbooster-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unifiedbooster
-Version: 0.2.2
+Version: 0.3.0
 Summary: Unified interface for Gradient Boosted Decision Trees
 Home-page: https://github.com/thierrymoudiki/unifiedbooster
 Author: T. Moudiki

unifiedbooster-0.3.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+unifiedbooster/__init__.py,sha256=3d8wQVXaeVIxqtk_STM6nvIGZiGTxKn9aAWjuwiDYuo,169
+unifiedbooster/gbdt.py,sha256=QCcWfXYfrOXdiSeygPEvVMjg9fVNjRaOnW9KsHK6bvo,4770
+unifiedbooster/gbdt_classification.py,sha256=UqZEOjDp_2hSm4jCxVoqz8vNQ-8JRW4Xn5CjFqPqRF4,4028
+unifiedbooster/gbdt_regression.py,sha256=ZNX5RJF-Wk2KJpOUD-lgNnqruDHZpzSTxdKeayv6iw0,3519
+unifiedbooster-0.3.0.dist-info/LICENSE,sha256=3rWw63btcdqbC0XMnpzCQhxDP8Vx7yKkKS7EDgJiY_4,1061
+unifiedbooster-0.3.0.dist-info/METADATA,sha256=E0drgIWtoGZNF1lkxrj_zlbMxq8QmOPIW4iDY_GPKm0,930
+unifiedbooster-0.3.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+unifiedbooster-0.3.0.dist-info/entry_points.txt,sha256=OVNTsCzMYnaJ11WIByB7G8Lym_dj-ERKZyQxWFUcW30,59
+unifiedbooster-0.3.0.dist-info/top_level.txt,sha256=gOMxxpRtx8_nJXTWsXJDFkNeCsjSJQPs6aUXKK5_nI4,15
+unifiedbooster-0.3.0.dist-info/RECORD,,

unifiedbooster-0.2.2.dist-info/RECORD

@@ -1,10 +0,0 @@
-unifiedbooster/__init__.py,sha256=3d8wQVXaeVIxqtk_STM6nvIGZiGTxKn9aAWjuwiDYuo,169
-unifiedbooster/gbdt.py,sha256=Y5gweEEfI5DGMsuE_Z-gJlbteawolzk2uFyoGXGwgKA,3958
-unifiedbooster/gbdt_classification.py,sha256=sIdoeqOBYGmzeHV3Rk3DrBav-ebG2v-o9cv0X0LREXg,3634
-unifiedbooster/gbdt_regression.py,sha256=VE129JJ_DNoi3knLI5I_5QNRkW0M3KAKRmE-sfXrCLk,3127
-unifiedbooster-0.2.2.dist-info/LICENSE,sha256=3rWw63btcdqbC0XMnpzCQhxDP8Vx7yKkKS7EDgJiY_4,1061
-unifiedbooster-0.2.2.dist-info/METADATA,sha256=Y-BX935CwMCs4uD-mNoniEeU0F0yuySg97dhZF5NSEQ,930
-unifiedbooster-0.2.2.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-unifiedbooster-0.2.2.dist-info/entry_points.txt,sha256=OVNTsCzMYnaJ11WIByB7G8Lym_dj-ERKZyQxWFUcW30,59
-unifiedbooster-0.2.2.dist-info/top_level.txt,sha256=gOMxxpRtx8_nJXTWsXJDFkNeCsjSJQPs6aUXKK5_nI4,15
-unifiedbooster-0.2.2.dist-info/RECORD,,