Package ml.dmlc.xgboost4j.java
Class XGBoostJNI
java.lang.Object
ml.dmlc.xgboost4j.java.XGBoostJNI
xgboost JNI functions
change 2015-7-6: *use a long[] (length=1) as container of handle to get the output DMatrix or Booster
-
Field Summary
Fields -
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescription(package private) static void
checkCall
(int ret) Check the return code of the JNI call.(package private) static final int
RabitAllreduce
(ByteBuffer sendrecvbuf, int count, int enum_dtype, int enum_op) static final int
static final int
RabitGetRank
(int[] out) static final int
RabitGetWorldSize
(int[] out) static final int
static final int
RabitTrackerPrint
(String msg) static final int
RabitVersionNumber
(int[] out) static final String
static final int
XGBoosterBoostOneIter
(long handle, long dtrain, float[] grad, float[] hess) static final int
XGBoosterCreate
(long[] handles, long[] out) static final int
XGBoosterDumpModelEx
(long handle, String fmap, int with_stats, String format, String[][] out_strings) static final int
XGBoosterDumpModelExWithFeatures
(long handle, String[] feature_names, int with_stats, String format, String[][] out_strings) static final int
XGBoosterEvalOneIter
(long handle, int iter, long[] dmats, String[] evnames, String[] eval_info) static final int
XGBoosterFree
(long handle) static final int
XGBoosterGetAttr
(long handle, String key, String[] out_string) static final int
XGBoosterGetAttrNames
(long handle, String[][] out_strings) static final int
XGBoosterGetModelRaw
(long handle, byte[][] out_bytes) static final int
XGBoosterLoadModel
(long handle, String fname) static final int
XGBoosterLoadModelFromBuffer
(long handle, byte[] bytes) static final int
XGBoosterLoadRabitCheckpoint
(long handle, int[] out_version) static final int
XGBoosterPredict
(long handle, long dmat, int option_mask, int ntree_limit, float[][] predicts) static final int
XGBoosterSaveModel
(long handle, String fname) static final int
XGBoosterSaveRabitCheckpoint
(long handle) static final int
XGBoosterSetAttr
(long handle, String key, String value) static final int
XGBoosterSetParam
(long handle, String name, String value) static final int
XGBoosterUpdateOneIter
(long handle, int iter, long dtrain) static final int
XGDMatrixCreateFromCSCEx
(long[] colptr, int[] indices, float[] data, int shapeParam, long[] out) static final int
XGDMatrixCreateFromCSREx
(long[] indptr, int[] indices, float[] data, int shapeParam, long[] out) (package private) static final int
XGDMatrixCreateFromDataIter
(Iterator<DataBatch> iter, String cache_info, long[] out) static final int
XGDMatrixCreateFromFile
(String fname, int silent, long[] out) static final int
XGDMatrixCreateFromMat
(float[] data, int nrow, int ncol, float missing, long[] out) static final int
XGDMatrixFree
(long handle) static final int
XGDMatrixGetFloatInfo
(long handle, String field, float[][] info) static final int
XGDMatrixGetUIntInfo
(long handle, String filed, int[][] info) static final int
XGDMatrixNumRow
(long handle, long[] row) static final int
XGDMatrixSaveBinary
(long handle, String fname, int silent) static final int
XGDMatrixSetFloatInfo
(long handle, String field, float[] array) static final int
XGDMatrixSetGroup
(long handle, int[] group) static final int
XGDMatrixSetUIntInfo
(long handle, String field, int[] array) static final int
XGDMatrixSliceDMatrix
(long handle, int[] idxset, long[] out)
-
Field Details
-
logger
private static final org.apache.commons.logging.Log logger
-
-
Constructor Details
-
XGBoostJNI
XGBoostJNI()
-
-
Method Details
-
checkCall
Check the return code of the JNI call.- Throws:
XGBoostError
- if the call failed.
-
XGBGetLastError
-
XGDMatrixCreateFromFile
-
XGDMatrixCreateFromDataIter
-
XGDMatrixCreateFromCSREx
public static final int XGDMatrixCreateFromCSREx(long[] indptr, int[] indices, float[] data, int shapeParam, long[] out) -
XGDMatrixCreateFromCSCEx
public static final int XGDMatrixCreateFromCSCEx(long[] colptr, int[] indices, float[] data, int shapeParam, long[] out) -
XGDMatrixCreateFromMat
public static final int XGDMatrixCreateFromMat(float[] data, int nrow, int ncol, float missing, long[] out) -
XGDMatrixSliceDMatrix
public static final int XGDMatrixSliceDMatrix(long handle, int[] idxset, long[] out) -
XGDMatrixFree
public static final int XGDMatrixFree(long handle) -
XGDMatrixSaveBinary
-
XGDMatrixSetFloatInfo
-
XGDMatrixSetUIntInfo
-
XGDMatrixSetGroup
public static final int XGDMatrixSetGroup(long handle, int[] group) -
XGDMatrixGetFloatInfo
-
XGDMatrixGetUIntInfo
-
XGDMatrixNumRow
public static final int XGDMatrixNumRow(long handle, long[] row) -
XGBoosterCreate
public static final int XGBoosterCreate(long[] handles, long[] out) -
XGBoosterFree
public static final int XGBoosterFree(long handle) -
XGBoosterSetParam
-
XGBoosterUpdateOneIter
public static final int XGBoosterUpdateOneIter(long handle, int iter, long dtrain) -
XGBoosterBoostOneIter
public static final int XGBoosterBoostOneIter(long handle, long dtrain, float[] grad, float[] hess) -
XGBoosterEvalOneIter
-
XGBoosterPredict
public static final int XGBoosterPredict(long handle, long dmat, int option_mask, int ntree_limit, float[][] predicts) -
XGBoosterLoadModel
-
XGBoosterSaveModel
-
XGBoosterLoadModelFromBuffer
public static final int XGBoosterLoadModelFromBuffer(long handle, byte[] bytes) -
XGBoosterGetModelRaw
public static final int XGBoosterGetModelRaw(long handle, byte[][] out_bytes) -
XGBoosterDumpModelEx
-
XGBoosterDumpModelExWithFeatures
-
XGBoosterGetAttrNames
-
XGBoosterGetAttr
-
XGBoosterSetAttr
-
XGBoosterLoadRabitCheckpoint
public static final int XGBoosterLoadRabitCheckpoint(long handle, int[] out_version) -
XGBoosterSaveRabitCheckpoint
public static final int XGBoosterSaveRabitCheckpoint(long handle) -
RabitInit
-
RabitFinalize
public static final int RabitFinalize() -
RabitTrackerPrint
-
RabitGetRank
public static final int RabitGetRank(int[] out) -
RabitGetWorldSize
public static final int RabitGetWorldSize(int[] out) -
RabitVersionNumber
public static final int RabitVersionNumber(int[] out) -
RabitAllreduce
-