dataframe-0.4.0.5: A fast, safe, and intuitive DataFrame library.
Safe HaskellNone
LanguageHaskell2010

DataFrame.DecisionTree

Documentation

data TreeConfig Source #

Constructors

TreeConfig 

Fields

Instances

Instances details
Show TreeConfig Source # 
Instance details

Defined in DataFrame.DecisionTree

Methods

showsPrec :: Int -> TreeConfig -> ShowS #

show :: TreeConfig -> String #

showList :: [TreeConfig] -> ShowS #

Eq TreeConfig Source # 
Instance details

Defined in DataFrame.DecisionTree

Methods

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

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

data SynthConfig Source #

Constructors

SynthConfig 

Fields

Instances

Instances details
Show SynthConfig Source # 
Instance details

Defined in DataFrame.DecisionTree

Methods

showsPrec :: Int -> SynthConfig -> ShowS #

show :: SynthConfig -> String #

showList :: [SynthConfig] -> ShowS #

Eq SynthConfig Source # 
Instance details

Defined in DataFrame.DecisionTree

Methods

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

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

defaultSynthConfig :: SynthConfig Source #

defaultTreeConfig :: TreeConfig Source #

fitDecisionTree :: Columnable a => TreeConfig -> Expr a -> DataFrame -> Expr a Source #

buildTree :: Columnable a => TreeConfig -> Int -> Text -> [Expr Bool] -> DataFrame -> Expr a Source #

pruneTree :: (Columnable a, Eq a) => Expr a -> Expr a Source #

type CondGen = SynthConfig -> DataFrame -> [Expr Bool] Source #

numericConditions :: CondGen Source #

generateNumericConds :: SynthConfig -> DataFrame -> [Expr Bool] Source #

numericExprsWithTerms :: SynthConfig -> DataFrame -> [Expr Double] Source #

numericCols :: DataFrame -> [Expr Double] Source #

numericExprs :: SynthConfig -> DataFrame -> [Expr Double] -> Int -> Int -> [Expr Double] Source #

boolExprs :: DataFrame -> [Expr Bool] -> [Expr Bool] -> Int -> Int -> [Expr Bool] Source #

generateConditionsOld :: SynthConfig -> DataFrame -> [Expr Bool] Source #

partitionDataFrame :: Expr Bool -> DataFrame -> (DataFrame, DataFrame) Source #

findBestSplit :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Maybe (Expr Bool) Source #

calculateGini :: Columnable a => Text -> DataFrame -> Double Source #

majorityValue :: Columnable a => Text -> DataFrame -> a Source #

getCounts :: Columnable a => Text -> DataFrame -> Map a Int Source #