hex.tree

Class SharedTreePojoWriter

java.lang.Object

hex.tree.SharedTreePojoWriter

All Implemented Interfaces:

hex.PojoWriter

public abstract class SharedTreePojoWriter
extends java.lang.Object
implements hex.PojoWriter

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	_binomialOpt
protected hex.genmodel.CategoricalEncoding	_encoding
protected water.Key<?>	_modelKey
protected hex.Model.Output	_output
protected CompressedTree[][]	_trees
protected TreeStats	_treeStats

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	SharedTreePojoWriter(water.Key<?> modelKey, hex.Model.Output output, hex.genmodel.CategoricalEncoding encoding, boolean binomialOpt, CompressedTree[][] trees, TreeStats treeStats)

Method Summary

Modifier and Type	Method and Description
boolean	toJavaCheckTooBig()
water.util.SBPrintStream	toJavaInit(water.util.SBPrintStream sb, water.codegen.CodeGeneratorPipeline fileContext)
void	toJavaPredictBody(water.util.SBPrintStream body, water.codegen.CodeGeneratorPipeline classCtx, water.codegen.CodeGeneratorPipeline fileCtx, boolean verboseCode)
protected abstract void	toJavaUnifyPreds(water.util.SBPrintStream body)

Methods inherited from class java.lang.Object

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

Methods inherited from interface hex.PojoWriter

toJavaTransform

Field Detail

_modelKey

protected final water.Key<?> _modelKey

_output

protected final hex.Model.Output _output

_encoding

protected final hex.genmodel.CategoricalEncoding _encoding

_binomialOpt

protected final boolean _binomialOpt

_trees

protected final CompressedTree[][] _trees

_treeStats

protected final TreeStats _treeStats

Constructor Detail

SharedTreePojoWriter

protected SharedTreePojoWriter(water.Key<?> modelKey,
                               hex.Model.Output output,
                               hex.genmodel.CategoricalEncoding encoding,
                               boolean binomialOpt,
                               CompressedTree[][] trees,
                               TreeStats treeStats)

Method Detail

toJavaCheckTooBig

public boolean toJavaCheckTooBig()

Specified by:

toJavaCheckTooBig in interface hex.PojoWriter

toJavaInit

public water.util.SBPrintStream toJavaInit(water.util.SBPrintStream sb,
                                           water.codegen.CodeGeneratorPipeline fileContext)

Specified by:

toJavaInit in interface hex.PojoWriter

toJavaPredictBody

public void toJavaPredictBody(water.util.SBPrintStream body,
                              water.codegen.CodeGeneratorPipeline classCtx,
                              water.codegen.CodeGeneratorPipeline fileCtx,
                              boolean verboseCode)

Specified by:

toJavaPredictBody in interface hex.PojoWriter

toJavaUnifyPreds

protected abstract void toJavaUnifyPreds(water.util.SBPrintStream body)