kumulant
com.eignex.kumulant/schema/spec/DecisionTreeClassifier

DecisionTreeClassifier

@Serializable
@SerialName(value = "DecisionTreeClassifier")
data class DecisionTreeClassifier(val featureSize: Int, val numClasses: Int, val splitCandidates: List<SerializableSplit>, val config: ClassificationTreeConfig = ClassificationTreeConfig(), val randomSeed: Int = StatDefaults.TREE_RANDOM_SEED, val leafArm: SeriesStatSpec<ClassCountsResult>? = null) : RegressionStatSpec<TreeClassificationResult> (source)

Spec for DecisionTreeClassifierStat: online VFDT classification tree.

Constructors

DecisionTreeClassifier

constructor(featureSize: Int, numClasses: Int, splitCandidates: List<SerializableSplit>, config: ClassificationTreeConfig = ClassificationTreeConfig(), randomSeed: Int = StatDefaults.TREE_RANDOM_SEED, leafArm: SeriesStatSpec<ClassCountsResult>? = null)(source)

Properties

config

val config: ClassificationTreeConfig(source)

RegressionTree growth tunables.

featureSize

val featureSize: Int(source)

Number of input features.

leafArm

val leafArm: SeriesStatSpec<ClassCountsResult>?(source)

Per-leaf accumulator recipe; null uses the stat's default leaf arm.

numClasses

val numClasses: Int(source)

Number of classes; y must round to [0, numClasses).

randomSeed

val randomSeed: Int(source)

PRNG seed.

splitCandidates

val splitCandidates: List<SerializableSplit>(source)

Candidate splits considered at every audit leaf.