diff --git a/Makefile b/Makefile index eeea8de..394cf4d 100644 --- a/Makefile +++ b/Makefile @@ -55,16 +55,16 @@ coverage: ## check code coverage quickly with the default Python docs: install ## generate docs pip install black pdoc - black nnetsauce/* --line-length=80 - find nnetsauce/ -name "*.py" -exec autopep8 --max-line-length=80 --in-place {} + - pdoc -t docs nnetsauce/* --output-dir nnetsauce-docs + black unifiedbooster/* --line-length=80 + find unifiedbooster/ -name "*.py" -exec autopep8 --max-line-length=80 --in-place {} + + pdoc -t docs unifiedbooster/* --output-dir unifiedbooster-docs find . -name '__pycache__' -exec rm -fr {} + servedocs: install ## compile the docs watching for change pip install black pdoc - black nnetsauce/* --line-length=80 - find nnetsauce/ -name "*.py" -exec autopep8 --max-line-length=80 --in-place {} + - pdoc -t docs nnetsauce/* + black unifiedbooster/* --line-length=80 + find unifiedbooster/ -name "*.py" -exec autopep8 --max-line-length=80 --in-place {} + + pdoc -t docs unifiedbooster/* find . -name '__pycache__' -exec rm -fr {} + build-site: docs ## export mkdocs website to a folder diff --git a/setup.py b/setup.py index 86172d1..58559f9 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ subprocess.check_call(['pip', 'install', 'Cython']) -__version__ = "0.2.0" +__version__ = "0.2.1" here = path.abspath(path.dirname(__file__)) diff --git a/unifiedbooster/gbdt.py b/unifiedbooster/gbdt.py index 9003ff6..c2c7d0e 100644 --- a/unifiedbooster/gbdt.py +++ b/unifiedbooster/gbdt.py @@ -55,6 +55,7 @@ def __init__(self, 'depth': self.max_depth, 'verbose': self.verbose, 'random_seed': self.seed, + 'bootstrap_type': 'Bernoulli', **kwargs }