org.jpmml.h2o

Class SharedTreeMojoModelConverter<M extends hex.genmodel.algos.tree.SharedTreeMojoModel>

java.lang.Object

org.jpmml.h2o.Converter<M>

org.jpmml.h2o.SharedTreeMojoModelConverter<M>

Direct Known Subclasses:

DrfMojoModelConverter, GbmMojoModelConverter, IsolationForestMojoModelConverter

public abstract class SharedTreeMojoModelConverter<M extends hex.genmodel.algos.tree.SharedTreeMojoModel>
extends Converter<M>

Constructor Summary

Constructors

Constructor and Description
SharedTreeMojoModelConverter(M model)

Method Summary

Modifier and Type	Method and Description
static org.dmg.pmml.tree.Node	encodeNode(SharedTree sharedTree, hex.genmodel.utils.ByteBufferWrapper byteBuffer, Integer id, org.dmg.pmml.Predicate predicate, org.jpmml.converter.CategoryManager categoryManager, org.jpmml.converter.PredicateManager predicateManager, org.jpmml.converter.Schema schema)
static org.dmg.pmml.Model	encodeTreeEnsemble(List<org.dmg.pmml.tree.TreeModel> treeModels, Function<List<org.dmg.pmml.tree.TreeModel>,org.dmg.pmml.mining.MiningModel> ensembleFunction)
static org.dmg.pmml.tree.TreeModel	encodeTreeModel(SharedTree sharedTree, org.jpmml.converter.PredicateManager predicateManager, org.jpmml.converter.Schema schema)
List<org.dmg.pmml.tree.TreeModel>	encodeTreeModels(org.jpmml.converter.Schema schema)
protected void	ensureRecordCount(org.dmg.pmml.tree.Node node, double recordCount)
protected void	ensureScore(org.dmg.pmml.tree.Node node, double score)
static byte[][]	getCompressedTrees(hex.genmodel.algos.tree.SharedTreeMojoModel model)
static byte[][]	getCompressedTreesAux(hex.genmodel.algos.tree.SharedTreeMojoModel model)
static int	getNTreeGroups(hex.genmodel.algos.tree.SharedTreeMojoModel model)
static int	getNTreesPerGroup(hex.genmodel.algos.tree.SharedTreeMojoModel model)

Methods inherited from class org.jpmml.h2o.Converter

encodeModel, encodePMML, encodeSchema, getDeclaredClass, getFieldValue, getFieldValue, getModel, toMojoModelSchema

Methods inherited from class java.lang.Object

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

Constructor Detail

SharedTreeMojoModelConverter

public SharedTreeMojoModelConverter(M model)

Method Detail

encodeTreeModels

public List<org.dmg.pmml.tree.TreeModel> encodeTreeModels(org.jpmml.converter.Schema schema)

ensureScore

protected void ensureScore(org.dmg.pmml.tree.Node node,
                           double score)

ensureRecordCount

protected void ensureRecordCount(org.dmg.pmml.tree.Node node,
                                 double recordCount)

encodeTreeModel

public static org.dmg.pmml.tree.TreeModel encodeTreeModel(SharedTree sharedTree,
                                                          org.jpmml.converter.PredicateManager predicateManager,
                                                          org.jpmml.converter.Schema schema)

encodeNode

public static org.dmg.pmml.tree.Node encodeNode(SharedTree sharedTree,
                                                hex.genmodel.utils.ByteBufferWrapper byteBuffer,
                                                Integer id,
                                                org.dmg.pmml.Predicate predicate,
                                                org.jpmml.converter.CategoryManager categoryManager,
                                                org.jpmml.converter.PredicateManager predicateManager,
                                                org.jpmml.converter.Schema schema)

encodeTreeEnsemble

public static org.dmg.pmml.Model encodeTreeEnsemble(List<org.dmg.pmml.tree.TreeModel> treeModels,
                                                    Function<List<org.dmg.pmml.tree.TreeModel>,org.dmg.pmml.mining.MiningModel> ensembleFunction)

getCompressedTrees

public static byte[][] getCompressedTrees(hex.genmodel.algos.tree.SharedTreeMojoModel model)

getCompressedTreesAux

public static byte[][] getCompressedTreesAux(hex.genmodel.algos.tree.SharedTreeMojoModel model)

getNTreeGroups

public static int getNTreeGroups(hex.genmodel.algos.tree.SharedTreeMojoModel model)

getNTreesPerGroup

public static int getNTreesPerGroup(hex.genmodel.algos.tree.SharedTreeMojoModel model)