hex.tree.xgboost

Class XGBoostUtils

java.lang.Object

hex.tree.xgboost.XGBoostUtils

public class XGBoostUtils
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
XGBoostUtils()

Method Summary

Modifier and Type	Method and Description
static hex.tree.xgboost.XGBoostUtils.FeatureProperties	assembleFeatureNames(hex.DataInfo di)
static ml.dmlc.xgboost4j.java.DMatrix	convertChunksToDMatrix(hex.DataInfo di, water.fvec.Chunk[] chunks, int response, boolean sparse, int offset) convert a set of H2O chunks (representing a part of a vector) to a sparse DMatrix
static ml.dmlc.xgboost4j.java.DMatrix	convertFrameToDMatrix(hex.DataInfo di, water.fvec.Frame frame, java.lang.String response, java.lang.String weight, java.lang.String offset, boolean sparse) convert an H2O Frame to a sparse DMatrix
static java.lang.String	makeFeatureMap(water.fvec.Frame f, hex.DataInfo di)

Methods inherited from class java.lang.Object

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

Constructor Detail

XGBoostUtils

public XGBoostUtils()

Method Detail

makeFeatureMap

public static java.lang.String makeFeatureMap(water.fvec.Frame f,
                                              hex.DataInfo di)

convertFrameToDMatrix

public static ml.dmlc.xgboost4j.java.DMatrix convertFrameToDMatrix(hex.DataInfo di,
                                                                   water.fvec.Frame frame,
                                                                   java.lang.String response,
                                                                   java.lang.String weight,
                                                                   java.lang.String offset,
                                                                   boolean sparse)
                                                            throws ml.dmlc.xgboost4j.java.XGBoostError

convert an H2O Frame to a sparse DMatrix

Parameters:

di - data info

frame - H2O Frame - adapted using a provided data info

response - name of the response column

weight - name of the weight column

Returns:

DMatrix

Throws:

ml.dmlc.xgboost4j.java.XGBoostError

convertChunksToDMatrix

public static ml.dmlc.xgboost4j.java.DMatrix convertChunksToDMatrix(hex.DataInfo di,
                                                                    water.fvec.Chunk[] chunks,
                                                                    int response,
                                                                    boolean sparse,
                                                                    int offset)
                                                             throws ml.dmlc.xgboost4j.java.XGBoostError

convert a set of H2O chunks (representing a part of a vector) to a sparse DMatrix

Returns:

DMatrix

Throws:

ml.dmlc.xgboost4j.java.XGBoostError

assembleFeatureNames

public static hex.tree.xgboost.XGBoostUtils.FeatureProperties assembleFeatureNames(hex.DataInfo di)