hex.tree.xgboost.matrix

Class SparseMatrixFactory

java.lang.Object

hex.tree.xgboost.matrix.SparseMatrixFactory

public class SparseMatrixFactory
extends java.lang.Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SparseMatrixFactory.SparseDMatrixProvider

Constructor Summary

Constructors

Constructor and Description
SparseMatrixFactory()

Method Summary

Modifier and Type	Method and Description
static SparseMatrix	allocateCSRMatrix(SparseMatrixDimensions sparseMatrixDimensions) Creates a SparseMatrix object with pre-instantiated backing arrays for row-oriented compression schema (CSR).
protected static SparseMatrixDimensions	calculateCSRMatrixDimensions(water.fvec.Chunk[] chunks, hex.DataInfo di, int weightColIndex)
static SparseMatrixDimensions	calculateCSRMatrixDimensions(water.fvec.Frame f, int[] chunkIds, water.fvec.Vec w, hex.DataInfo di)
static ai.h2o.xgboost4j.java.DMatrix	csr(water.fvec.Chunk[] chunks, int weight, int respIdx, int offsetIdx, hex.DataInfo di, float[] resp, float[] weights, float[] offsets)
static MatrixLoader.DMatrixProvider	csr(water.fvec.Frame frame, int[] chunksIds, water.fvec.Vec weightsVec, water.fvec.Vec offsetsVec, water.fvec.Vec responseVec, hex.DataInfo di, float[] resp, float[] weights, float[] offsets)
static int	initializeFromChunkIds(water.fvec.Frame frame, int[] chunks, water.fvec.Vec weightsVec, water.fvec.Vec offsetsVec, hex.DataInfo di, SparseMatrix matrix, SparseMatrixDimensions dimensions, water.fvec.Vec respVec, float[] resp, float[] weights, float[] offsets)
static SparseMatrixFactory.SparseDMatrixProvider	toDMatrix(SparseMatrix sm, SparseMatrixDimensions smd, int actualRows, int shape, float[] resp, float[] weights, float[] offsets)

Methods inherited from class java.lang.Object

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

Constructor Detail

SparseMatrixFactory

public SparseMatrixFactory()

Method Detail

csr

public static MatrixLoader.DMatrixProvider csr(water.fvec.Frame frame,
                                               int[] chunksIds,
                                               water.fvec.Vec weightsVec,
                                               water.fvec.Vec offsetsVec,
                                               water.fvec.Vec responseVec,
                                               hex.DataInfo di,
                                               float[] resp,
                                               float[] weights,
                                               float[] offsets)

csr

public static ai.h2o.xgboost4j.java.DMatrix csr(water.fvec.Chunk[] chunks,
                                                int weight,
                                                int respIdx,
                                                int offsetIdx,
                                                hex.DataInfo di,
                                                float[] resp,
                                                float[] weights,
                                                float[] offsets)
                                         throws ai.h2o.xgboost4j.java.XGBoostError

Throws:

ai.h2o.xgboost4j.java.XGBoostError

toDMatrix

public static SparseMatrixFactory.SparseDMatrixProvider toDMatrix(SparseMatrix sm,
                                                                  SparseMatrixDimensions smd,
                                                                  int actualRows,
                                                                  int shape,
                                                                  float[] resp,
                                                                  float[] weights,
                                                                  float[] offsets)

initializeFromChunkIds

public static int initializeFromChunkIds(water.fvec.Frame frame,
                                         int[] chunks,
                                         water.fvec.Vec weightsVec,
                                         water.fvec.Vec offsetsVec,
                                         hex.DataInfo di,
                                         SparseMatrix matrix,
                                         SparseMatrixDimensions dimensions,
                                         water.fvec.Vec respVec,
                                         float[] resp,
                                         float[] weights,
                                         float[] offsets)

allocateCSRMatrix

public static SparseMatrix allocateCSRMatrix(SparseMatrixDimensions sparseMatrixDimensions)

Creates a SparseMatrix object with pre-instantiated backing arrays for row-oriented compression schema (CSR). All backing arrays are allocated using MemoryManager.

Parameters:

sparseMatrixDimensions - Dimensions of a sparse matrix

Returns:

An instance of SparseMatrix with pre-allocated backing arrays.

calculateCSRMatrixDimensions

protected static SparseMatrixDimensions calculateCSRMatrixDimensions(water.fvec.Chunk[] chunks,
                                                                     hex.DataInfo di,
                                                                     int weightColIndex)

calculateCSRMatrixDimensions

public static SparseMatrixDimensions calculateCSRMatrixDimensions(water.fvec.Frame f,
                                                                  int[] chunkIds,
                                                                  water.fvec.Vec w,
                                                                  hex.DataInfo di)