Spark ML -- Decision Trees
Perform regression or classification using decision trees.
ml_decision_tree(x, response, features, impurity = c("auto", "gini", "entropy", "variance"), max.bins = 32L, max.depth = 5L, min.info.gain = 0, min.rows = 1L, type = c("auto", "regression", "classification"), thresholds = NULL, seed = NULL, checkpoint.interval = 10L, cache.node.ids = FALSE, max.memory = 256L, ml.options = ml_options(), ...)
An object coercable to a Spark DataFrame (typically, a
The name of the response vector (as a length-one character
vector), or a formula, giving a symbolic description of the model to be
The name of features (terms) to use for the model fit.
Criterion used for information gain calculation One of 'auto', 'gini', 'entropy', or 'variance'. 'auto' defaults to 'gini' for classification and 'variance' for regression.
The maximum number of bins used for discretizing continuous features and for choosing how to split on features at each node. More bins give higher granularity.
Maximum depth of the tree (>= 0); that is, the maximum number of nodes separating any leaves from the root of the tree.
Minimum information gain for a split to be considered at a tree node. Should be >= 0, defaults to 0.
Minimum number of instances each child must have after split.
The type of model to fit.
Thresholds in multi-class classification to adjust the probability of predicting each class. Vector must have length equal to the number of classes, with values > 0 excepting that at most one value may be 0. The class with largest value p/t is predicted, where p is the original probability of that class and t is the class's threshold.
Seed for random numbers.
Set checkpoint interval (>= 1) or disable checkpoint (-1). E.g. 10 means that the cache will get checkpointed every 10 iterations, defaults to 10.
Maximum memory in MB allocated to histogram aggregation. If too small, then 1 node will be split per iteration, and its aggregates may exceed this size. Defaults to 256.
Optional arguments, used to affect the model generated. See
Optional arguments. The
Other Spark ML routines: