Method to train a decision tree model.
Method to train a decision tree model. The method supports binary and multiclass classification and regression.
Note: Using org.apache.spark.mllib.tree.DecisionTree$#trainClassifier and org.apache.spark.mllib.tree.DecisionTree$#trainRegressor is recommended to clearly separate classification and regression.
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint. For classification, labels should take values {0, 1, ..., numClasses-1}. For regression, labels are real numbers.
Type of decision tree, either classification or regression.
Criterion used for information gain calculation.
Maximum depth of the tree (e.g. depth 0 means 1 leaf node, depth 1 means 1 internal node + 2 leaf nodes).
Number of classes for classification. Default value of 2.
Maximum number of bins used for splitting features.
Algorithm for calculating quantiles.
Map storing arity of categorical features. An entry (n -> k) indicates that feature n is categorical with k categories indexed from 0: {0, 1, ..., k-1}.
DecisionTreeModel that can be used for prediction.
Method to train a decision tree model.
Method to train a decision tree model. The method supports binary and multiclass classification and regression.
Note: Using org.apache.spark.mllib.tree.DecisionTree$#trainClassifier and org.apache.spark.mllib.tree.DecisionTree$#trainRegressor is recommended to clearly separate classification and regression.
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint. For classification, labels should take values {0, 1, ..., numClasses-1}. For regression, labels are real numbers.
Type of decision tree, either classification or regression.
Criterion used for information gain calculation.
Maximum depth of the tree (e.g. depth 0 means 1 leaf node, depth 1 means 1 internal node + 2 leaf nodes).
Number of classes for classification. Default value of 2.
DecisionTreeModel that can be used for prediction.
Method to train a decision tree model.
Method to train a decision tree model. The method supports binary and multiclass classification and regression.
Note: Using org.apache.spark.mllib.tree.DecisionTree$#trainClassifier and org.apache.spark.mllib.tree.DecisionTree$#trainRegressor is recommended to clearly separate classification and regression.
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint. For classification, labels should take values {0, 1, ..., numClasses-1}. For regression, labels are real numbers.
Type of decision tree, either classification or regression.
Criterion used for information gain calculation.
Maximum depth of the tree (e.g. depth 0 means 1 leaf node, depth 1 means 1 internal node + 2 leaf nodes).
DecisionTreeModel that can be used for prediction.
Method to train a decision tree model.
Method to train a decision tree model. The method supports binary and multiclass classification and regression.
Note: Using org.apache.spark.mllib.tree.DecisionTree$#trainClassifier and org.apache.spark.mllib.tree.DecisionTree$#trainRegressor is recommended to clearly separate classification and regression.
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint. For classification, labels should take values {0, 1, ..., numClasses-1}. For regression, labels are real numbers.
The configuration parameters for the tree algorithm which specify the type of decision tree (classification or regression), feature type (continuous, categorical), depth of the tree, quantile calculation strategy, etc.
DecisionTreeModel that can be used for prediction.
Java-friendly API for org.apache.spark.mllib.tree.DecisionTree$#trainClassifier
Java-friendly API for org.apache.spark.mllib.tree.DecisionTree$#trainClassifier
Method to train a decision tree model for binary or multiclass classification.
Method to train a decision tree model for binary or multiclass classification.
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint. Labels should take values {0, 1, ..., numClasses-1}.
Number of classes for classification.
Map storing arity of categorical features. An entry (n -> k) indicates that feature n is categorical with k categories indexed from 0: {0, 1, ..., k-1}.
Criterion used for information gain calculation. Supported values: "gini" (recommended) or "entropy".
Maximum depth of the tree (e.g. depth 0 means 1 leaf node, depth 1 means 1 internal node + 2 leaf nodes). (suggested value: 5)
Maximum number of bins used for splitting features. (suggested value: 32)
DecisionTreeModel that can be used for prediction.
Java-friendly API for org.apache.spark.mllib.tree.DecisionTree$#trainRegressor
Java-friendly API for org.apache.spark.mllib.tree.DecisionTree$#trainRegressor
Method to train a decision tree model for regression.
Method to train a decision tree model for regression.
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint. Labels are real numbers.
Map storing arity of categorical features. An entry (n -> k) indicates that feature n is categorical with k categories indexed from 0: {0, 1, ..., k-1}.
Criterion used for information gain calculation. The only supported value for regression is "variance".
Maximum depth of the tree (e.g. depth 0 means 1 leaf node, depth 1 means 1 internal node + 2 leaf nodes). (suggested value: 5)
Maximum number of bins used for splitting features. (suggested value: 32)
DecisionTreeModel that can be used for prediction.