diff --git a/quantile_forest/_quantile_forest.py b/quantile_forest/_quantile_forest.py index 5dc2049..c030ca2 100755 --- a/quantile_forest/_quantile_forest.py +++ b/quantile_forest/_quantile_forest.py @@ -1,5 +1,4 @@ -""" -Quantile regression forest of trees-based ensemble methods. +"""Quantile regression forest of trees-based ensemble methods. The module structure is the following: @@ -859,11 +858,12 @@ class RandomForestQuantileRegressor(BaseForestQuantileRegressor): """A random forest regressor that provides quantile estimates. A quantile random forest is a meta estimator that fits a number of - decision trees on various sub-samples of the dataset and uses averaging - to improve the predictive accuracy and control over-fitting. The - sub-sample size is controlled with the `max_samples` parameter if - `bootstrap=True` (default), otherwise the whole dataset is used to build - each tree. + decision trees on various sub-samples of the dataset, keeps the values of + samples that reach each node, and assesses the conditional distribution + based on this information. The sub-sample size is controlled with the + `max_samples` parameter if `bootstrap=True` (default), otherwise the whole + dataset is used to build each tree. The leaf-sample size is controlled + with the `max_samples_leaf` parameter. Parameters ---------- @@ -1146,10 +1146,11 @@ def _more_tags(self): class ExtraTreesQuantileRegressor(BaseForestQuantileRegressor): """An extra-trees regressor that provides quantile estimates. - This class implements a meta estimator that fits a number of randomized - decision trees (a.k.a. extra-trees) on various sub-samples of the dataset - and use averaging to improve the predictive accuracy and control - over-fitting. + A quantile extra trees regressor is a meta estimator that fits a number of + randomized decision trees (a.k.a. extra-trees) on various sub-samples of + the dataset, keeps the values of samples that reach each node, and + assesses the conditional distribution based on this information. The + leaf-sample size is controlled with the `max_samples_leaf` parameter. Parameters ---------- diff --git a/quantile_forest/_quantile_forest_fast.pxd b/quantile_forest/_quantile_forest_fast.pxd index cfbb6d5..3f6a403 100755 --- a/quantile_forest/_quantile_forest_fast.pxd +++ b/quantile_forest/_quantile_forest_fast.pxd @@ -12,12 +12,12 @@ ctypedef double float64_t cdef class QuantileForest: # The QuantileForest object. - # Input/Output layout + # Input/Output layout. cdef public vector[vector[float64_t]] y_train cdef public intp_t[:, :, :, :] y_train_leaves cdef public bint sparse_pickle - # Methods + # Methods. cpdef cnp.ndarray predict( self, vector[double] quantiles, diff --git a/quantile_forest/tests/test_quantile_forest.py b/quantile_forest/tests/test_quantile_forest.py index a5ff72e..02ea480 100755 --- a/quantile_forest/tests/test_quantile_forest.py +++ b/quantile_forest/tests/test_quantile_forest.py @@ -1,6 +1,4 @@ -""" -Testing for the quantile forest module (quantile_forest._quantile_forest). -""" +"""Testing for the quantile forest module (quantile_forest._quantile_forest).""" import math import pickle diff --git a/quantile_forest/version.txt b/quantile_forest/version.txt index 4bd55fe..60c0c5d 100644 --- a/quantile_forest/version.txt +++ b/quantile_forest/version.txt @@ -1 +1 @@ -1.3.dev +1.3.dev0