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.

Input ports

1. Type: Spark Data
   Input data
   Input Spark DataFrame/RDD
2. Type: PMML
   Mapping dictionary
   PMML with the nominal values mapping dictionary (optional)

Output ports

1. Type: Spark MLlib Model
   Gradient Boosted Trees Model
   Spark MLlib Gradient-Boosted Trees Model