DecisionTreeClassifier
@Serializable
@SerialName(value = "DecisionTreeClassifier")
Spec for DecisionTreeClassifierStat: online VFDT classification tree.
Constructors
DecisionTreeClassifier
constructor(featureSize: Int, numClasses: Int, splitCandidates: List<SerializableSplit>, config: ClassificationTreeConfig = ClassificationTreeConfig(), randomSeed: Int = 0)(source)
Properties
config
RegressionTree growth tunables.
featureSize
Number of input features.
numClasses
Number of classes; y must round to [0, numClasses).
randomSeed
PRNG seed.
splitCandidates
Candidate splits considered at every audit leaf.