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 |
---|---|
|
HTML representation of estimator. This is redundant with the logic of _repr_mimebundle_. The latter should … |
|
Return the feature importances. The importance of a feature is computed as the (normalized) total reduction … |
|
DEPRECATED: The attribute n_features_ is deprecated in 1.0 and will be removed in 1.2. Use n_features_in_ instead. |
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)#
Bases:
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)#
- _abc_impl = <_abc._abc_data object>#
- _fit_reglin(X, y, sample_weight)#
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.
- _mapping_train(X)#
- _predict_reglin(X, check_input=True)#
Computes the predictions with a linear regression fitted with the observations mapped to each leave of the tree.
- Parameters:
X – array-like 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, array of shape = [n_samples] or [n_samples, n_outputs] The predicted classes, or the predict values.
- fit(X, y, sample_weight=None, check_input=True)#
Replaces the string stored in criterion by an instance of a class.
- predict(X, check_input=True)#
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.
- predict_leaves(X)#
Returns the leave index for each observation of X.
- Parameters:
X – array
- Returns:
array leaves index in
self.leaves_index_