Package ml.dmlc.xgboost4j.java

Class JniUtils

java.lang.Object

ml.dmlc.xgboost4j.java.JniUtils

public final class JniUtils
extends Object

DJL class that has access to XGBoost JNI.

Method Summary

Modifier and Type	Method	Description
static void	checkCall(int ret)
static long	createDMatrix(Buffer buf, ai.djl.ndarray.types.Shape shape, float missing)
static long	createDMatrix(ml.dmlc.xgboost4j.java.ColumnBatch columnBatch, float missing, int nthread)
static long	createDMatrixCSR(long[] indptr, int[] indices, float[] array, float missing, int nthread)
static void	deleteDMatrix(long handle)
static void	deleteModel(long handle)
static float[]	inference(XgbSymbolBlock block, XgbNDArray array, int treeLimit, XgbSymbolBlock.Mode mode)
static XgbSymbolBlock	loadModel(XgbNDManager manager, String modelPath)

Methods inherited from class java.lang.Object

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

Method Details

checkCall

public static void checkCall(int ret)

loadModel

public static XgbSymbolBlock loadModel(XgbNDManager manager,
 String modelPath)

createDMatrix

public static long createDMatrix(Buffer buf,
 ai.djl.ndarray.types.Shape shape,
 float missing)

createDMatrix

public static long createDMatrix(ml.dmlc.xgboost4j.java.ColumnBatch columnBatch,
 float missing,
 int nthread)

createDMatrixCSR

public static long createDMatrixCSR(long[] indptr,
 int[] indices,
 float[] array,
 float missing,
 int nthread)

deleteDMatrix

public static void deleteDMatrix(long handle)

inference

public static float[] inference(XgbSymbolBlock block,
 XgbNDArray array,
 int treeLimit,
 XgbSymbolBlock.Mode mode)

deleteModel

public static void deleteModel(long handle)