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 Spark DataFrame/RDD
  2. Type: PMML
    PMML with the nominal values mapping dictionary (optional)

Output Ports

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


This node is part of the extension

KNIME Extension for Apache Spark


Short Link

Drag node into KNIME Analytics Platform