hivemall.xgboost.utils

Class SparseDMatrixBuilder

java.lang.Object

hivemall.xgboost.utils.DMatrixBuilder

hivemall.xgboost.utils.SparseDMatrixBuilder

public final class SparseDMatrixBuilder
extends DMatrixBuilder

Constructor Summary

Constructors

Constructor and Description
SparseDMatrixBuilder(int initSize)
SparseDMatrixBuilder(int initSize, boolean sortRequired)

Method Summary

Modifier and Type	Method and Description
ml.dmlc.xgboost4j.java.DMatrix	buildMatrix(float[] labels)
SparseDMatrixBuilder	nextColumn(int col, float value)
SparseDMatrixBuilder	nextRow()

Methods inherited from class hivemall.xgboost.utils.DMatrixBuilder

checkColIndex, nextColumn, nextRow, nextRow, nextRow

Methods inherited from class java.lang.Object

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

Constructor Detail

SparseDMatrixBuilder

public SparseDMatrixBuilder(@Nonnegative
                            int initSize)

SparseDMatrixBuilder

public SparseDMatrixBuilder(@Nonnegative
                            int initSize,
                            boolean sortRequired)

Method Detail

nextRow

@Nonnull
public SparseDMatrixBuilder nextRow()

Specified by:

nextRow in class DMatrixBuilder

nextColumn

@Nonnull
public SparseDMatrixBuilder nextColumn(@Nonnegative
                                                int col,
                                                float value)

Specified by:

nextColumn in class DMatrixBuilder

buildMatrix

@Nonnull
public ml.dmlc.xgboost4j.java.DMatrix buildMatrix(@Nonnull
                                                           float[] labels)
                                                    throws ml.dmlc.xgboost4j.java.XGBoostError

Specified by:

buildMatrix in class DMatrixBuilder

Throws:

ml.dmlc.xgboost4j.java.XGBoostError