Safe Haskell	None
Language	Haskell2010

DataFrame.DecisionTree.Regression

Description

Variance-reduction (weighted-SSE) regression trees over the CART feature machinery; leaves predict the weighted mean of their rows. fitRegTreeOn lets gradient boosting refit on residuals without re-extracting features.

Synopsis

data RegTreeConfig = RegTreeConfig {
- rtMaxDepth :: !Int
- rtMinSamplesSplit :: !Int
- rtMinLeafSize :: !Int
- rtMinImpurityDecrease :: !Double
}
defaultRegTreeConfig :: RegTreeConfig
fitRegTreeOn :: RegTreeConfig -> Vector CartFeature -> Vector Double -> Maybe (Vector Double) -> Tree Double

Documentation

data RegTreeConfig Source #

Stopping criteria for the regression tree.

Constructors

RegTreeConfig
Fields rtMaxDepth :: !Int rtMinSamplesSplit :: !Int rtMinLeafSize :: !Int rtMinImpurityDecrease :: !Double

Instances

Instances details

Show RegTreeConfig Source #

Instance details

Defined in DataFrame.DecisionTree.Regression

Methods

showsPrec :: Int -> RegTreeConfig -> ShowS #

show :: RegTreeConfig -> String #

showList :: [RegTreeConfig] -> ShowS #

Eq RegTreeConfig Source #

Instance details

Defined in DataFrame.DecisionTree.Regression

Methods

(==) :: RegTreeConfig -> RegTreeConfig -> Bool #

(/=) :: RegTreeConfig -> RegTreeConfig -> Bool #

Fit RegTreeConfig (Expr Double) Source #

Instance details

Defined in DataFrame.DecisionTree.Model

Associated Types

type ModelOf RegTreeConfig (Expr Double)
Instance details Defined in DataFrame.DecisionTree.Model type ModelOf RegTreeConfig (Expr Double) = DecisionTreeRegressor
type FrameReq RegTreeConfig (Expr Double)
Instance details Defined in DataFrame.DecisionTree.Model type FrameReq RegTreeConfig (Expr Double) = 'AnyFrame

Methods

fit :: RegTreeConfig -> Expr Double -> FrameFor (Expr Double) -> FitResult (FrameFor (Expr Double)) (ModelOf RegTreeConfig (Expr Double)) Source #

type FrameReq RegTreeConfig (Expr Double) Source #

Instance details

Defined in DataFrame.DecisionTree.Model

type FrameReq RegTreeConfig (Expr Double) = 'AnyFrame

type ModelOf RegTreeConfig (Expr Double) Source #

Instance details

Defined in DataFrame.DecisionTree.Model

type ModelOf RegTreeConfig (Expr Double) = DecisionTreeRegressor

defaultRegTreeConfig :: RegTreeConfig Source #

Implementation verb used by the fit/predict instances and boosting.

fitRegTreeOn :: RegTreeConfig -> Vector CartFeature -> Vector Double -> Maybe (Vector Double) -> Tree Double Source #

Fit on pre-extracted features, a target vector, and optional per-row weights (length n). Used by gradient boosting on residual targets.