packages feed

dataframe-learn-2.5.0.0: src/DataFrame/DecisionTree.hs

-- | Interpretable decision trees on DataFrames (CART refined by TAO).
module DataFrame.DecisionTree (
    -- * Estimators

    -- | Fitted-model records with their Fit and Predict instances.
    module DataFrame.DecisionTree.Model,

    -- * Classifier configuration
    TreeConfig (..),
    defaultTreeConfig,
    SynthConfig (..),
    defaultSynthConfig,
    ColumnOrdering (..),
    orderable,
    defaultColumnOrdering,
    withOrdFrom,

    -- * Regressor configuration
    RegTreeConfig (..),
    defaultRegTreeConfig,

    -- * Fitted tree structure
    Tree (..),
    treeDepth,

    -- * Solver configuration (fills @TreeConfig.linearSolverConfig@)
    SolverConfig (..),
    defaultSolverConfig,
) where

import DataFrame.DecisionTree.Model
import DataFrame.DecisionTree.Regression (
    RegTreeConfig (..),
    defaultRegTreeConfig,
 )
import DataFrame.DecisionTree.Types (
    ColumnOrdering (..),
    SynthConfig (..),
    Tree (..),
    TreeConfig (..),
    defaultColumnOrdering,
    defaultSynthConfig,
    defaultTreeConfig,
    orderable,
    treeDepth,
    withOrdFrom,
 )
import DataFrame.LinearSolver (SolverConfig (..), defaultSolverConfig)