hex.tree.xgboost

Class XGBoostPojoWriter

java.lang.Object

hex.tree.xgboost.XGBoostPojoWriter

public abstract class XGBoostPojoWriter
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
protected java.lang.String	_namePrefix
protected XGBoostOutput	_output
protected biz.k11i.xgboost.Predictor	_p

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	XGBoostPojoWriter(biz.k11i.xgboost.Predictor p, java.lang.String namePrefix, XGBoostOutput output, double defaultThreshold)

Method Summary

Modifier and Type	Method and Description
protected java.lang.String	getFeatureAccessor(int idx)
static XGBoostPojoWriter	make(biz.k11i.xgboost.Predictor p, java.lang.String namePrefix, XGBoostOutput output, double defaultThreshold)
protected abstract void	renderComputePredict(water.util.SBPrintStream sb, water.codegen.CodeGeneratorPipeline fileCtx)
void	renderJavaPredictBody(water.util.SBPrintStream sb, water.codegen.CodeGeneratorPipeline fileCtx)

Methods inherited from class java.lang.Object

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

Field Detail

_p

protected final biz.k11i.xgboost.Predictor _p

_namePrefix

protected final java.lang.String _namePrefix

_output

protected final XGBoostOutput _output

Constructor Detail

XGBoostPojoWriter

protected XGBoostPojoWriter(biz.k11i.xgboost.Predictor p,
                            java.lang.String namePrefix,
                            XGBoostOutput output,
                            double defaultThreshold)

Method Detail

make

public static XGBoostPojoWriter make(biz.k11i.xgboost.Predictor p,
                                     java.lang.String namePrefix,
                                     XGBoostOutput output,
                                     double defaultThreshold)

getFeatureAccessor

protected java.lang.String getFeatureAccessor(int idx)

renderJavaPredictBody

public void renderJavaPredictBody(water.util.SBPrintStream sb,
                                  water.codegen.CodeGeneratorPipeline fileCtx)

renderComputePredict

protected abstract void renderComputePredict(water.util.SBPrintStream sb,
                                             water.codegen.CodeGeneratorPipeline fileCtx)