unifiedbooster 0.1.0__py3-none-any.whl

unifiedbooster/__init__.py

@@ -0,0 +1,4 @@
+from .gbdt_classification import GBDTClassifier
+from .gbdt_regression import GBDTRegressor
+
+__all__ = ["GBDTClassifier", "GBDTRegressor"]

unifiedbooster/gbdt_classification.py

@@ -0,0 +1,80 @@
+from sklearn.base import BaseEstimator, ClassifierMixin
+from xgboost import XGBClassifier
+from catboost import CatBoostClassifier
+from lightgbm import LGBMClassifier
+
+
+class GBDTClassifier(BaseEstimator, ClassifierMixin):
+    def __init__(self, model_type='xgboost', 
+                 n_estimators=100, 
+                 learning_rate=0.1, 
+                 max_depth=3, 
+                 subsample=1.0,  
+                 verbosity=0,
+                 **kwargs):
+        self.model_type = model_type
+        self.n_estimators = n_estimators
+        self.learning_rate = learning_rate
+        self.max_depth = max_depth
+        self.subsample = subsample
+        self.verbosity = verbosity
+        # xgboost -----
+        # n_estimators        
+        # learning_rate
+        # subsample
+        # max_depth
+        # lightgbm -----
+        # num_iterations
+        # learning_rate
+        # bagging_fraction
+        # max_depth        
+        # catboost -----
+        # iterations
+        # learning_rate        
+        # rsm 
+        # depth
+        if self.model_type == "xgboost": 
+            self.params = {
+                'n_estimators': self.n_estimators,
+                'learning_rate': self.learning_rate,
+                'subsample': self.subsample,
+                'max_depth': self.max_depth,
+                'verbosity': self.verbosity,
+                **kwargs
+            }
+        elif self.model_type == "lightgbm":
+             self.params = {
+                'num_iterations': self.n_estimators,
+                'learning_rate': self.learning_rate,
+                'bagging_fraction': self.subsample,
+                'max_depth': self.max_depth,
+                'verbose': self.verbosity - 1 if self.verbosity==0 else self.verbosity
+                **kwargs
+            }
+        elif self.model_type == "catboost":
+             self.params = {
+                'iterations': self.n_estimators,
+                'learning_rate': self.learning_rate,
+                'rsm': self.subsample,
+                'depth': self.max_depth,
+                'verbose': self.verbosity
+                **kwargs
+            }           
+        
+        if model_type == 'xgboost':
+            self.model = XGBClassifier(**self.params)
+        elif model_type == 'catboost':            
+            self.model = CatBoostClassifier(**self.params, logging_level='Silent')
+        elif model_type == 'lightgbm':
+            self.model = LGBMClassifier(**self.params)
+        else:
+            raise ValueError(f"Unknown model_type: {model_type}")
+    
+    def fit(self, X, y, **kwargs):
+        return self.model.fit(X, y, **kwargs)
+    
+    def predict(self, X):
+        return self.model.predict(X)
+    
+    def predict_proba(self, X):
+        return self.model.predict_proba(X)

unifiedbooster/gbdt_regression.py

@@ -0,0 +1,77 @@
+from sklearn.base import BaseEstimator, RegressorMixin
+from xgboost import XGBRegressor
+from catboost import CatBoostRegressor
+from lightgbm import LGBMRegressor
+
+
+class GBDTRegressor(BaseEstimator, RegressorMixin):
+    def __init__(self, model_type='xgboost', 
+                 n_estimators=100, 
+                 learning_rate=0.1, 
+                 max_depth=3, 
+                 subsample=1.0,    
+                 verbosity=0,              
+                 **kwargs):
+        self.model_type = model_type
+        self.n_estimators = n_estimators
+        self.learning_rate = learning_rate
+        self.max_depth = max_depth
+        self.subsample = subsample
+        self.verbosity = verbosity
+        # xgboost -----
+        # n_estimators        
+        # learning_rate
+        # subsample
+        # max_depth
+        # lightgbm -----
+        # num_iterations
+        # learning_rate
+        # bagging_fraction
+        # max_depth        
+        # catboost -----
+        # iterations
+        # learning_rate        
+        # rsm 
+        # depth
+        if self.model_type == "xgboost": 
+            self.params = {
+                'n_estimators': self.n_estimators,
+                'learning_rate': self.learning_rate,
+                'subsample': self.subsample,
+                'max_depth': self.max_depth,
+                'verbosity': self.verbosity,
+                **kwargs
+            }
+        elif self.model_type == "lightgbm":
+             self.params = {
+                'num_iterations': self.n_estimators,
+                'learning_rate': self.learning_rate,
+                'bagging_fraction': self.subsample,
+                'max_depth': self.max_depth,
+                'verbose': self.verbosity - 1 if self.verbosity==0 else self.verbosity
+                **kwargs
+            }
+        elif self.model_type == "catboost":
+             self.params = {
+                'iterations': self.n_estimators,
+                'learning_rate': self.learning_rate,
+                'rsm': self.subsample,
+                'depth': self.max_depth,
+                'verbose': self.verbosity
+                **kwargs
+            }           
+        
+        if model_type == 'xgboost':
+            self.model = XGBRegressor(**self.params)
+        elif model_type == 'catboost':            
+            self.model = CatBoostRegressor(**self.params, logging_level='Silent')
+        elif model_type == 'lightgbm':
+            self.model = LGBMRegressor(**self.params)
+        else:
+            raise ValueError(f"Unknown model_type: {model_type}")
+    
+    def fit(self, X, y, **kwargs):
+        return self.model.fit(X, y, **kwargs)
+    
+    def predict(self, X):
+        return self.model.predict(X)        

unifiedbooster-0.1.0.dist-info/METADATA

@@ -0,0 +1,25 @@
+Metadata-Version: 2.1
+Name: unifiedbooster
+Version: 0.1.0
+Summary: Call R functions from Python
+Home-page: https://github.com/thierrymoudiki/unifiedbooster
+Author: T. Moudiki
+Author-email: thierry.moudiki@gmail.com
+License: BSD license
+Keywords: unifiedbooster
+Classifier: Development Status :: 2 - Pre-Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.6
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Requires-Python: >=3.6
+Requires-Dist: Cython
+Requires-Dist: scikit-learn
+Requires-Dist: xgboost
+Requires-Dist: lightgbm
+Requires-Dist: catboost
+
+Unified interface for Gradient Boosted Decision Trees

unifiedbooster-0.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+unifiedbooster/__init__.py,sha256=urmH2DAkJ4e-idIh0oMZX9lP9rlqTPyZ-qHuNeJcEIc,137
+unifiedbooster/gbdt_classification.py,sha256=4KIPuPGdXpQvppgklkS_7tB180WTAuxZKHC8ulCJjQs,2784
+unifiedbooster/gbdt_regression.py,sha256=BrM51nKDkd9mU3TnmRqT4QLaQLfrQ_inQEIJvmhj1PQ,2719
+unifiedbooster-0.1.0.dist-info/METADATA,sha256=8ATMu6Xyb5ab8B0Og5nlakWnZ0kGu6OitZiDG23QuXc,841
+unifiedbooster-0.1.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+unifiedbooster-0.1.0.dist-info/entry_points.txt,sha256=OVNTsCzMYnaJ11WIByB7G8Lym_dj-ERKZyQxWFUcW30,59
+unifiedbooster-0.1.0.dist-info/top_level.txt,sha256=gOMxxpRtx8_nJXTWsXJDFkNeCsjSJQPs6aUXKK5_nI4,15
+unifiedbooster-0.1.0.dist-info/RECORD,,

unifiedbooster-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (72.1.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

unifiedbooster-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+unifiedbooster = unifiedbooster.cli:main

unifiedbooster-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+unifiedbooster