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 ml.dmlc.xgboost4j.java.DMatrix	convertChunksToDMatrix(water.Key<hex.DataInfo> dataInfoKey, water.fvec.Chunk[] chunks, int response, int weight, int fold, boolean sparse) convert a set of H2O chunks (representing a part of a vector) to a sparse DMatrix
static ml.dmlc.xgboost4j.java.DMatrix	convertFrameToDMatrix(water.Key<hex.DataInfo> dataInfoKey, water.fvec.Frame f, boolean onlyLocal, java.lang.String response, java.lang.String weight, java.lang.String fold, java.lang.String[] featureMap, boolean sparse) convert an H2O Frame to a sparse DMatrix

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

convertFrameToDMatrix

public static ml.dmlc.xgboost4j.java.DMatrix convertFrameToDMatrix(water.Key<hex.DataInfo> dataInfoKey,
                                                                   water.fvec.Frame f,
                                                                   boolean onlyLocal,
                                                                   java.lang.String response,
                                                                   java.lang.String weight,
                                                                   java.lang.String fold,
                                                                   java.lang.String[] featureMap,
                                                                   boolean sparse)
                                                            throws ml.dmlc.xgboost4j.java.XGBoostError

convert an H2O Frame to a sparse DMatrix

Parameters:

f - H2O Frame

onlyLocal - if true uses only chunks local to this node

response - name of the response column

weight - name of the weight column

fold - name of the fold assignment column

featureMap - featureMap[0] will be populated with the column names and types

Returns:

DMatrix

Throws:

ml.dmlc.xgboost4j.java.XGBoostError

convertChunksToDMatrix

public static ml.dmlc.xgboost4j.java.DMatrix convertChunksToDMatrix(water.Key<hex.DataInfo> dataInfoKey,
                                                                    water.fvec.Chunk[] chunks,
                                                                    int response,
                                                                    int weight,
                                                                    int fold,
                                                                    boolean sparse)
                                                             throws ml.dmlc.xgboost4j.java.XGBoostError

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

Parameters:

response - name of the response column

weight - name of the weight column

fold - name of the fold assignment column

Returns:

DMatrix

Throws:

ml.dmlc.xgboost4j.java.XGBoostError