50% off | Unlimited Data & AI Learning
Get 50% off unlimited learning

hhcartr (version 1.0.0)

hhcartr_regressor_find_better_split: hhcartr_regressor_find_better_split finds the best feature to split on.

Description

This internal function is used by the regressor model to find the feature column that will offer the best split based on sum of squares.

Usage

hhcartr_regressor_find_better_split(X, y, max_features)

Arguments

X

feature variables to search for the best split.

y

target variable.

max_features

the maximum number of features to use when splitting a node

Value

a list of the following variables (var_idx_, split_, score_)