module mlmodel.piecewise_tree_regression
¶
Short summary¶
module mlinsights.mlmodel.piecewise_tree_regression
Implements a kind of piecewise linear regression by modifying the criterion used by the algorithm which builds a decision tree.
Classes¶
class 
truncated documentation 

Implements a kind of piecewise linear regression by modifying the criterion used by the algorithm which builds a decision … 
Properties¶
property 
truncated documentation 


Return the feature importances. The importance of a feature is computed as the (normalized) total reduction … 
Methods¶
method 
truncated documentation 

Fits linear regressions for all leaves. Sets attributes 

Computes the predictions with a linear regression fitted with the observations mapped to each leave of the … 

Replaces the string stored in criterion by an instance of a class. 

Overloads method predict. Falls back into the predict from a decision tree is criterion is mse, mae, … 

Returns the leave index for each observation of X. 
Documentation¶
Implements a kind of piecewise linear regression by modifying the criterion used by the algorithm which builds a decision tree.

class
mlinsights.mlmodel.piecewise_tree_regression.
PiecewiseTreeRegressor
(criterion='mselin', splitter='best', max_depth=None, min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0.0, max_features=None, random_state=None, max_leaf_nodes=None, min_impurity_decrease=0.0, min_impurity_split=None, presort=False)[source]¶ Bases:
sklearn.tree.tree.DecisionTreeRegressor
Implements a kind of piecewise linear regression by modifying the criterion used by the algorithm which builds a decision tree. See sklearn.tree.DecisionTreeRegressor to get the meaning of the parameters except criterion:
mselin
: optimizes for a piecewise linear regressionsimple
: optimizes for a stepwise regression (equivalent to mse)

__abstractmethods__
= frozenset({})¶

__init__
(criterion='mselin', splitter='best', max_depth=None, min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0.0, max_features=None, random_state=None, max_leaf_nodes=None, min_impurity_decrease=0.0, min_impurity_split=None, presort=False)[source]¶ Initialize self. See help(type(self)) for accurate signature.

_abc_impl
= <_abc_data object>¶

_fit_reglin
(X, y, sample_weight)[source]¶ Fits linear regressions for all leaves. Sets attributes
leaves_mapping_
,betas_
,leaves_index_
. The first attribute is a dictionary{leave: row}
which maps a leave of the tree to the coefficientsbetas_[row, :]
of a regression trained on all training points mapped a specific leave.leaves_index_
keeps in memory a set of leaves.

_predict_reglin
(X, check_input=True)[source]¶ Computes the predictions with a linear regression fitted with the observations mapped to each leave of the tree.
 Parameters
X (arraylike or sparse matrix of shape = [n_samples, n_features]) – The input samples. Internally, it will be converted to
dtype=np.float32
and if a sparse matrix is provided to a sparsecsr_matrix
.check_input (boolean, (default=True)) – Allow to bypass several input checking. Don’t use this parameter unless you know what you do.
 Returns
y – The predicted classes, or the predict values.
 Return type
array of shape = [n_samples] or [n_samples, n_outputs]

fit
(X, y, sample_weight=None, check_input=True, X_idx_sorted=None)[source]¶ Replaces the string stored in criterion by an instance of a class.

predict
(X, check_input=True)[source]¶ Overloads method predict. Falls back into the predict from a decision tree is criterion is mse, mae, simple. Computes the predictions from linear regression if the criterion is mselin.