Skip to content

Commit b682361

Browse files
committed
feat: show and eq instances of decision tree configs.
1 parent 5fea971 commit b682361

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

src/DataFrame/DecisionTree.hs

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,15 @@ import Type.Reflection (typeRep)
3636

3737
import DataFrame.Functions ((.<), (.<=), (.==), (.>), (.>=))
3838

39-
data TreeConfig = TreeConfig
40-
{ maxTreeDepth :: Int
41-
, minSamplesSplit :: Int
42-
, minLeafSize :: Int
43-
, synthConfig :: SynthConfig
44-
}
39+
data TreeConfig
40+
= TreeConfig
41+
{ maxTreeDepth :: Int
42+
, minSamplesSplit :: Int
43+
, minLeafSize :: Int
44+
, synthConfig :: SynthConfig
45+
}
46+
derivng
47+
(Eq, Show)
4548

4649
data SynthConfig = SynthConfig
4750
{ maxExprDepth :: Int
@@ -52,6 +55,7 @@ data SynthConfig = SynthConfig
5255
, enableCrossCols :: Bool
5356
, enableArithOps :: Bool
5457
}
58+
deriving (Eq, Show)
5559

5660
defaultSynthConfig :: SynthConfig
5761
defaultSynthConfig =
@@ -68,7 +72,7 @@ defaultSynthConfig =
6872
defaultTreeConfig :: TreeConfig
6973
defaultTreeConfig =
7074
TreeConfig
71-
{ maxTreeDepth = 10
75+
{ maxTreeDepth = 4
7276
, minSamplesSplit = 5
7377
, minLeafSize = 1
7478
, synthConfig = defaultSynthConfig

0 commit comments

Comments
 (0)