hivemall.xgboost

Class XGBoostTrainUDTF

java.lang.Object

org.apache.hadoop.hive.ql.udf.generic.GenericUDTF

hivemall.UDTFWithOptions

hivemall.xgboost.XGBoostTrainUDTF

public class XGBoostTrainUDTF
extends UDTFWithOptions

UDTF for train_xgboost

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	XGBoostTrainUDTF.ObjectiveType

Field Summary

Fields

Modifier and Type	Field and Description
protected int	numClass
protected XGBoostTrainUDTF.ObjectiveType	objectiveType
protected Map<String,Object>	params

Fields inherited from class hivemall.UDTFWithOptions

mapredContext

Constructor Summary

Constructors

Constructor and Description
XGBoostTrainUDTF()

Method Summary

Modifier and Type	Method and Description
void	close()
protected org.apache.commons.cli.Options	getOptions()
org.apache.hadoop.hive.serde2.objectinspector.StructObjectInspector	initialize(org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector[] argOIs)
protected void	onFinishTraining(ml.dmlc.xgboost4j.java.Booster booster)
void	process(Object[] args)
protected org.apache.commons.cli.CommandLine	processOptions(org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector[] argOIs)
protected float	processTargetValue(float target) To validate target range, overrides this method

Methods inherited from class hivemall.UDTFWithOptions

assumeFalse, assumeTrue, checkNotNull, checkNotNull, configure, getReporter, incrCounter, nonNullArgument, parseFeatures, parseOptions, reportProgress, setCounterValue, showHelp

Methods inherited from class org.apache.hadoop.hive.ql.udf.generic.GenericUDTF

forward, initialize, setCollector

Methods inherited from class java.lang.Object

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

Field Detail

params

@Nonnull
protected final Map<String,Object> params

numClass

protected int numClass

objectiveType

protected XGBoostTrainUDTF.ObjectiveType objectiveType

Constructor Detail

XGBoostTrainUDTF

public XGBoostTrainUDTF()

Method Detail

getOptions

protected org.apache.commons.cli.Options getOptions()

Specified by:

getOptions in class UDTFWithOptions

processOptions

@Nonnull
protected org.apache.commons.cli.CommandLine processOptions(org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector[] argOIs)
                                                              throws UDFArgumentException

Specified by:

processOptions in class UDTFWithOptions

Throws:

UDFArgumentException

initialize

public org.apache.hadoop.hive.serde2.objectinspector.StructObjectInspector initialize(@Nonnull
                                                                                      org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector[] argOIs)
                                                                               throws UDFArgumentException

Overrides:

initialize in class org.apache.hadoop.hive.ql.udf.generic.GenericUDTF

Throws:

UDFArgumentException

processTargetValue

protected float processTargetValue(float target)
                            throws org.apache.hadoop.hive.ql.metadata.HiveException

To validate target range, overrides this method

Throws:

org.apache.hadoop.hive.ql.metadata.HiveException

process

public void process(@Nonnull
                    Object[] args)
             throws org.apache.hadoop.hive.ql.metadata.HiveException

Specified by:

process in class org.apache.hadoop.hive.ql.udf.generic.GenericUDTF

Throws:

org.apache.hadoop.hive.ql.metadata.HiveException

close

public void close()
           throws org.apache.hadoop.hive.ql.metadata.HiveException

Specified by:

close in class org.apache.hadoop.hive.ql.udf.generic.GenericUDTF

Throws:

org.apache.hadoop.hive.ql.metadata.HiveException

onFinishTraining

@VisibleForTesting
protected void onFinishTraining(@Nonnull
                                                   ml.dmlc.xgboost4j.java.Booster booster)