org.jpmml.sparkml.model

Class TreeModelUtil

java.lang.Object

org.jpmml.sparkml.model.TreeModelUtil

public class TreeModelUtil
extends Object

Method Summary

Methods

Modifier and Type	Method and Description
static org.dmg.pmml.TreeModel	encodeDecisionTree(org.apache.spark.ml.tree.DecisionTreeModel model, org.jpmml.converter.Schema schema)
static List<org.dmg.pmml.TreeModel>	encodeDecisionTreeEnsemble(org.apache.spark.ml.tree.TreeEnsembleModel model, org.jpmml.converter.Schema schema)
static org.dmg.pmml.Node	encodeNode(org.dmg.pmml.MiningFunctionType miningFunction, org.apache.spark.ml.tree.Node node, org.jpmml.converter.Schema schema)
static org.dmg.pmml.TreeModel	encodeTreeModel(org.dmg.pmml.MiningFunctionType miningFunction, org.apache.spark.ml.tree.Node node, org.jpmml.converter.Schema schema)
static void	scalePredictions(org.dmg.pmml.TreeModel treeModel, double weight)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

encodeDecisionTree

public static org.dmg.pmml.TreeModel encodeDecisionTree(org.apache.spark.ml.tree.DecisionTreeModel model,
                                        org.jpmml.converter.Schema schema)

encodeDecisionTreeEnsemble

public static List<org.dmg.pmml.TreeModel> encodeDecisionTreeEnsemble(org.apache.spark.ml.tree.TreeEnsembleModel model,
                                                      org.jpmml.converter.Schema schema)

encodeTreeModel

public static org.dmg.pmml.TreeModel encodeTreeModel(org.dmg.pmml.MiningFunctionType miningFunction,
                                     org.apache.spark.ml.tree.Node node,
                                     org.jpmml.converter.Schema schema)

scalePredictions

public static void scalePredictions(org.dmg.pmml.TreeModel treeModel,
                    double weight)

encodeNode

public static org.dmg.pmml.Node encodeNode(org.dmg.pmml.MiningFunctionType miningFunction,
                           org.apache.spark.ml.tree.Node node,
                           org.jpmml.converter.Schema schema)