org.jpmml.sparkml.model

Class TreeModelUtil

java.lang.Object

org.jpmml.sparkml.model.TreeModelUtil

public class TreeModelUtil
extends Object

Method Summary

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

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.tree.TreeModel encodeDecisionTree(org.apache.spark.ml.tree.DecisionTreeModel model,
                                                             org.jpmml.converter.Schema schema)

encodeDecisionTreeEnsemble

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

encodeTreeModel

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

encodeNode

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