Spark Gradient-Boosted Trees Learner (MLlib)
This node applies the Apache Spark Gradient-Boosted Trees (GBTs) algorithm.
Note: GBTs do not yet support multiclass classification. The class column needs to contain 0 and 1 as class labels. For multiclass problems, please use decision trees or Random Forests.
Please note that all data must be numeric, including the label column which needs to contain 0 and 1. Use the Spark Category To Number nodes to convert nominal values to numeric columns. The mapping dictionary of the converter node is the input of the second in port of this node.
Use the Spark Predictor node to apply the learned model to unseen data.
- Type: Spark Data Input Spark DataFrame/RDD
- Type: PMML PMML with the nominal values mapping dictionary (optional)
- Type: Spark MLlib Model Spark MLlib Gradient-Boosted Trees Model