Safe Haskell	None
Language	Haskell2010

DataFrame.DecisionTree

Synopsis

data TreeConfig = TreeConfig {
- maxTreeDepth :: Int
- minSamplesSplit :: Int
- minLeafSize :: Int
- percentiles :: [Int]
- expressionPairs :: Int
- synthConfig :: SynthConfig
- taoIterations :: Int
- taoConvergenceTol :: Double
}
data SynthConfig = SynthConfig {
- maxExprDepth :: Int
- boolExpansion :: Int
- disallowedCombinations :: [(Text, Text)]
- complexityPenalty :: Double
- enableStringOps :: Bool
- enableCrossCols :: Bool
- enableArithOps :: Bool
}
defaultSynthConfig :: SynthConfig
defaultTreeConfig :: TreeConfig
data Tree a
- = Leaf !a
- | Branch !(Expr Bool) !(Tree a) !(Tree a)
treeDepth :: Tree a -> Int
treeToExpr :: Columnable a => Tree a -> Expr a
fitDecisionTree :: Columnable a => TreeConfig -> Expr a -> DataFrame -> Expr a
taoOptimize :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Vector Int -> Tree a -> Tree a
taoIteration :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Vector Int -> Tree a -> Tree a
optimizeDepthLevel :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Vector Int -> Tree a -> Int -> Tree a
optimizeAtDepth :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Vector Int -> Tree a -> Int -> Int -> Tree a
optimizeNode :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Vector Int -> Tree a -> Tree a
findBestSplitTAO :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Vector Int -> Tree a -> Tree a -> Expr Bool -> Expr Bool
data CarePoint = CarePoint {
- cpIndex :: !Int
- cpCorrectDir :: !Direction
}
data Direction
- = GoLeft
- | GoRight
identifyCarePoints :: Columnable a => Text -> DataFrame -> Vector Int -> Tree a -> Tree a -> [CarePoint]
predictWithTree :: Columnable a => Text -> DataFrame -> Int -> Tree a -> a
countCarePointErrors :: Expr Bool -> DataFrame -> [CarePoint] -> Int
partitionIndices :: Expr Bool -> DataFrame -> Vector Int -> (Vector Int, Vector Int)
majorityValueFromIndices :: Columnable a => Text -> DataFrame -> Vector Int -> a
computeTreeLoss :: Columnable a => Text -> DataFrame -> Vector Int -> Tree a -> Double
pruneDead :: Tree a -> Tree a
pruneExpr :: (Columnable a, Eq a) => Expr a -> Expr a
buildGreedyTree :: Columnable a => TreeConfig -> Int -> Text -> [Expr Bool] -> DataFrame -> Tree a
findBestGreedySplit :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Maybe (Expr Bool)
numericConditions :: TreeConfig -> DataFrame -> [Expr Bool]
generateNumericConds :: TreeConfig -> DataFrame -> [Expr Bool]
numericExprsWithTerms :: SynthConfig -> DataFrame -> [Expr Double]
numericCols :: DataFrame -> [Expr Double]
numericExprs :: SynthConfig -> DataFrame -> [Expr Double] -> Int -> Int -> [Expr Double]
boolExprs :: DataFrame -> [Expr Bool] -> [Expr Bool] -> Int -> Int -> [Expr Bool]
generateConditionsOld :: TreeConfig -> DataFrame -> [Expr Bool]
partitionDataFrame :: Expr Bool -> DataFrame -> (DataFrame, DataFrame)
calculateGini :: Columnable a => Text -> DataFrame -> Double
majorityValue :: Columnable a => Text -> DataFrame -> a
getCounts :: Columnable a => Text -> DataFrame -> Map a Int
percentile :: Int -> Expr Double -> DataFrame -> Double
buildTree :: Columnable a => TreeConfig -> Int -> Text -> [Expr Bool] -> DataFrame -> Expr a
findBestSplit :: Columnable a => TreeConfig -> Text -> [Expr Bool] -> DataFrame -> Maybe (Expr Bool)
pruneTree :: (Columnable a, Eq a) => Expr a -> Expr a

Documentation

data TreeConfig Source #

Constructors

TreeConfig
Fields maxTreeDepth :: Int minSamplesSplit :: Int minLeafSize :: Int percentiles :: [Int] expressionPairs :: Int synthConfig :: SynthConfig taoIterations :: Int taoConvergenceTol :: Double

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 maxExprDepth :: Int boolExpansion :: Int disallowedCombinations :: [(Text, Text)] complexityPenalty :: Double enableStringOps :: Bool enableCrossCols :: Bool enableArithOps :: Bool

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 #

data Tree a Source #

Constructors

Leaf !a
Branch !(Expr Bool) !(Tree a) !(Tree a)

Instances

Instances details

Show a => Show (Tree a) Source #
Instance details Defined in DataFrame.DecisionTree Methods showsPrec :: Int -> Tree a -> ShowS # show :: Tree a -> String # showList :: [Tree a] -> ShowS #
Eq a => Eq (Tree a) Source #
Instance details Defined in DataFrame.DecisionTree Methods (==) :: Tree a -> Tree a -> Bool # (/=) :: Tree a -> Tree a -> Bool #

treeDepth :: Tree a -> Int Source #

treeToExpr :: Columnable a => Tree a -> Expr a Source #

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

Fit a TAO decision tree

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

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

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

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

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

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

data CarePoint Source #

A care point with its index and which direction leads to correct classification

Constructors

CarePoint
Fields cpIndex :: !Int cpCorrectDir :: !Direction

Instances

Instances details

Show CarePoint Source #
Instance details Defined in DataFrame.DecisionTree Methods showsPrec :: Int -> CarePoint -> ShowS # show :: CarePoint -> String # showList :: [CarePoint] -> ShowS #
Eq CarePoint Source #
Instance details Defined in DataFrame.DecisionTree Methods (==) :: CarePoint -> CarePoint -> Bool # (/=) :: CarePoint -> CarePoint -> Bool #

data Direction Source #

Constructors

GoLeft
GoRight

Instances

Instances details

Show Direction Source #
Instance details Defined in DataFrame.DecisionTree Methods showsPrec :: Int -> Direction -> ShowS # show :: Direction -> String # showList :: [Direction] -> ShowS #
Eq Direction Source #
Instance details Defined in DataFrame.DecisionTree Methods (==) :: Direction -> Direction -> Bool # (/=) :: Direction -> Direction -> Bool #

identifyCarePoints :: Columnable a => Text -> DataFrame -> Vector Int -> Tree a -> Tree a -> [CarePoint] Source #

Identify care points: points where exactly one subtree classifies correctly

For each point reaching the node: 1. Compute what label the left subtree would predict 2. Compute what label the right subtree would predict 3. If exactly one matches the true label, it's a care point 4. Record which direction leads to correct classification

predictWithTree :: Columnable a => Text -> DataFrame -> Int -> Tree a -> a Source #

Predict the label for a single point using a fixed tree

countCarePointErrors :: Expr Bool -> DataFrame -> [CarePoint] -> Int Source #

partitionIndices :: Expr Bool -> DataFrame -> Vector Int -> (Vector Int, Vector Int) Source #

majorityValueFromIndices :: Columnable a => Text -> DataFrame -> Vector Int -> a Source #

computeTreeLoss :: Columnable a => Text -> DataFrame -> Vector Int -> Tree a -> Double Source #

pruneDead :: Tree a -> Tree a Source #

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

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

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