hivemall.xgboost.utils

Class XGBoostUtils

java.lang.Object

hivemall.xgboost.utils.XGBoostUtils

public final class XGBoostUtils
extends Object

Method Summary

Modifier and Type	Method and Description
static void	close(ml.dmlc.xgboost4j.java.Booster booster)
static void	close(ml.dmlc.xgboost4j.java.DMatrix matrix)
static ml.dmlc.xgboost4j.java.Booster	createBooster(ml.dmlc.xgboost4j.java.DMatrix matrix, Map<String,Object> params)
static ml.dmlc.xgboost4j.java.DMatrix	createDMatrix(List<XGBoostBatchPredictUDTF.LabeledPointWithRowId> data)
static ml.dmlc.xgboost4j.java.Booster	deserializeBooster(org.apache.hadoop.io.Text model)
static String	getVersion()
static biz.k11i.xgboost.Predictor	loadPredictor(org.apache.hadoop.io.Text model)
static biz.k11i.xgboost.util.FVec	parseRowAsFVec(String[] row, int start, int end)
static org.apache.hadoop.io.Text	serializeBooster(ml.dmlc.xgboost4j.java.Booster booster)

Methods inherited from class java.lang.Object

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

Method Detail

getVersion

@Nonnull
public static String getVersion()
                                  throws org.apache.hadoop.hive.ql.metadata.HiveException

Throws:

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

createDMatrix

@Nonnull
public static ml.dmlc.xgboost4j.java.DMatrix createDMatrix(@Nonnull
                                                                    List<XGBoostBatchPredictUDTF.LabeledPointWithRowId> data)
                                                             throws ml.dmlc.xgboost4j.java.XGBoostError

Throws:

ml.dmlc.xgboost4j.java.XGBoostError

createBooster

@Nonnull
public static ml.dmlc.xgboost4j.java.Booster createBooster(@Nonnull
                                                                    ml.dmlc.xgboost4j.java.DMatrix matrix,
                                                                    @Nonnull
                                                                    Map<String,Object> params)
                                                             throws NoSuchMethodException,
                                                                    ml.dmlc.xgboost4j.java.XGBoostError,
                                                                    IllegalAccessException,
                                                                    InvocationTargetException,
                                                                    InstantiationException

Throws:

NoSuchMethodException

ml.dmlc.xgboost4j.java.XGBoostError

IllegalAccessException

InvocationTargetException

InstantiationException

close

public static void close(@Nullable
                         ml.dmlc.xgboost4j.java.DMatrix matrix)

close

public static void close(@Nullable
                         ml.dmlc.xgboost4j.java.Booster booster)

serializeBooster

@Nonnull
public static org.apache.hadoop.io.Text serializeBooster(@Nonnull
                                                                  ml.dmlc.xgboost4j.java.Booster booster)
                                                           throws org.apache.hadoop.hive.ql.metadata.HiveException

Throws:

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

deserializeBooster

@Nonnull
public static ml.dmlc.xgboost4j.java.Booster deserializeBooster(@Nonnull
                                                                         org.apache.hadoop.io.Text model)
                                                                  throws org.apache.hadoop.hive.ql.metadata.HiveException

Throws:

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

loadPredictor

@Nonnull
public static biz.k11i.xgboost.Predictor loadPredictor(@Nonnull
                                                                org.apache.hadoop.io.Text model)
                                                         throws org.apache.hadoop.hive.ql.metadata.HiveException

Throws:

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

parseRowAsFVec

@Nonnull
public static biz.k11i.xgboost.util.FVec parseRowAsFVec(@Nonnull
                                                                 String[] row,
                                                                 int start,
                                                                 int end)