unifiedbooster 0.1.0__tar.gz

unifiedbooster-0.1.0/PKG-INFO

@@ -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/README.md

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

unifiedbooster-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

unifiedbooster-0.1.0/setup.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python
+
+"""The setup script."""
+
+import platform
+import subprocess
+from setuptools import setup, find_packages
+from codecs import open
+from os import path
+
+subprocess.check_call(['pip', 'install', 'Cython'])
+
+__version__ = "0.1.0"
+
+here = path.abspath(path.dirname(__file__))
+
+# get the dependencies and installs
+
+with open(
+    path.join(here, "requirements.txt"), encoding="utf-8"
+) as f:
+    all_reqs = f.read().split("\n")
+
+install_requires = [
+    x.strip() for x in all_reqs if "git+" not in x
+]
+dependency_links = [
+    x.strip().replace("git+", "")
+    for x in all_reqs
+    if x.startswith("git+")
+]
+
+setup(
+    author="T. Moudiki",
+    author_email='thierry.moudiki@gmail.com',
+    python_requires='>=3.6',
+    classifiers=[
+        'Development Status :: 2 - Pre-Alpha',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: BSD License',
+        'Natural Language :: English',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
+    ],
+    description="Call R functions from Python",
+    entry_points={
+        'console_scripts': [
+            'unifiedbooster=unifiedbooster.cli:main',
+        ],
+    },
+    install_requires=install_requires,
+    license="BSD license",
+    long_description="Unified interface for Gradient Boosted Decision Trees",
+    include_package_data=True,
+    keywords='unifiedbooster',
+    name='unifiedbooster',
+    packages=find_packages(include=['unifiedbooster', 'unifiedbooster.*']),
+    test_suite='tests',
+    url='https://github.com/thierrymoudiki/unifiedbooster',
+    version=__version__,
+    zip_safe=False,
+)

unifiedbooster-0.1.0/unifiedbooster/__init__.py

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

unifiedbooster-0.1.0/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-0.1.0/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/unifiedbooster.egg-info/PKG-INFO

@@ -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/unifiedbooster.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+setup.py
+unifiedbooster/__init__.py
+unifiedbooster/gbdt_classification.py
+unifiedbooster/gbdt_regression.py
+unifiedbooster.egg-info/PKG-INFO
+unifiedbooster.egg-info/SOURCES.txt
+unifiedbooster.egg-info/dependency_links.txt
+unifiedbooster.egg-info/entry_points.txt
+unifiedbooster.egg-info/not-zip-safe
+unifiedbooster.egg-info/requires.txt
+unifiedbooster.egg-info/top_level.txt

unifiedbooster-0.1.0/unifiedbooster.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

unifiedbooster-0.1.0/unifiedbooster.egg-info/entry_points.txt

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

unifiedbooster-0.1.0/unifiedbooster.egg-info/not-zip-safe

@@ -0,0 +1 @@
+

unifiedbooster-0.1.0/unifiedbooster.egg-info/requires.txt

@@ -0,0 +1,5 @@
+Cython
+scikit-learn
+xgboost
+lightgbm
+catboost

unifiedbooster-0.1.0/unifiedbooster.egg-info/top_level.txt

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