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 <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.DecisionTreeModel> org.dmg.pmml.tree.TreeModel	encodeDecisionTree(M model, org.jpmml.converter.PredicateManager predicateManager, org.jpmml.converter.Schema schema)
static <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.DecisionTreeModel> org.dmg.pmml.tree.TreeModel	encodeDecisionTree(M model, org.jpmml.converter.Schema schema)
static <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.TreeEnsembleModel<T>,T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.tree.DecisionTreeModel> List<org.dmg.pmml.tree.TreeModel>	encodeDecisionTreeEnsemble(M model, org.jpmml.converter.PredicateManager predicateManager, org.jpmml.converter.Schema schema)
static <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.TreeEnsembleModel<T>,T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.tree.DecisionTreeModel> List<org.dmg.pmml.tree.TreeModel>	encodeDecisionTreeEnsemble(M model, org.jpmml.converter.Schema schema)
static org.dmg.pmml.tree.Node	encodeNode(org.apache.spark.ml.tree.Node node, org.jpmml.converter.PredicateManager predicateManager, Map<org.dmg.pmml.FieldName,Set<String>> parentFieldValues, org.dmg.pmml.MiningFunction miningFunction, org.jpmml.converter.Schema schema)
static org.dmg.pmml.tree.TreeModel	encodeTreeModel(org.apache.spark.ml.tree.Node node, org.jpmml.converter.PredicateManager predicateManager, org.dmg.pmml.MiningFunction miningFunction, 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

encodeDecisionTreeEnsemble

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

encodeDecisionTreeEnsemble

public static <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.TreeEnsembleModel<T>,T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.tree.DecisionTreeModel> List<org.dmg.pmml.tree.TreeModel> encodeDecisionTreeEnsemble(M model,
                                                                                                                                                                                                                                                        org.jpmml.converter.PredicateManager predicateManager,
                                                                                                                                                                                                                                                        org.jpmml.converter.Schema schema)

encodeDecisionTree

public static <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.DecisionTreeModel> org.dmg.pmml.tree.TreeModel encodeDecisionTree(M model,
                                                                                                                                                   org.jpmml.converter.Schema schema)

encodeDecisionTree

public static <M extends org.apache.spark.ml.Model<M> & org.apache.spark.ml.tree.DecisionTreeModel> org.dmg.pmml.tree.TreeModel encodeDecisionTree(M model,
                                                                                                                                                   org.jpmml.converter.PredicateManager predicateManager,
                                                                                                                                                   org.jpmml.converter.Schema schema)

encodeTreeModel

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

encodeNode

public static org.dmg.pmml.tree.Node encodeNode(org.apache.spark.ml.tree.Node node,
                                                org.jpmml.converter.PredicateManager predicateManager,
                                                Map<org.dmg.pmml.FieldName,Set<String>> parentFieldValues,
                                                org.dmg.pmml.MiningFunction miningFunction,
                                                org.jpmml.converter.Schema schema)