Class XGBoostJNI

java.lang.Object
ml.dmlc.xgboost4j.java.XGBoostJNI

class XGBoostJNI extends Object
xgboost JNI functions change 2015-7-6: *use a long[] (length=1) as container of handle to get the output DMatrix or Booster
  • Field Details

    • logger

      private static final org.apache.commons.logging.Log logger
  • Constructor Details

    • XGBoostJNI

      XGBoostJNI()
  • Method Details

    • checkCall

      static void checkCall(int ret) throws XGBoostError
      Check the return code of the JNI call.
      Throws:
      XGBoostError - if the call failed.
    • XGBGetLastError

      public static final String XGBGetLastError()
    • XGDMatrixCreateFromFile

      public static final int XGDMatrixCreateFromFile(String fname, int silent, long[] out)
    • XGDMatrixCreateFromDataIter

      static final int XGDMatrixCreateFromDataIter(Iterator<DataBatch> iter, String cache_info, long[] out)
    • 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

      public static final int XGDMatrixSaveBinary(long handle, String fname, int silent)
    • XGDMatrixSetFloatInfo

      public static final int XGDMatrixSetFloatInfo(long handle, String field, float[] array)
    • XGDMatrixSetUIntInfo

      public static final int XGDMatrixSetUIntInfo(long handle, String field, int[] array)
    • XGDMatrixSetGroup

      public static final int XGDMatrixSetGroup(long handle, int[] group)
    • XGDMatrixGetFloatInfo

      public static final int XGDMatrixGetFloatInfo(long handle, String field, float[][] info)
    • XGDMatrixGetUIntInfo

      public static final int XGDMatrixGetUIntInfo(long handle, String filed, int[][] info)
    • 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

      public static final int XGBoosterSetParam(long handle, String name, String value)
    • 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

      public static final int XGBoosterEvalOneIter(long handle, int iter, long[] dmats, String[] evnames, String[] eval_info)
    • XGBoosterPredict

      public static final int XGBoosterPredict(long handle, long dmat, int option_mask, int ntree_limit, float[][] predicts)
    • XGBoosterLoadModel

      public static final int XGBoosterLoadModel(long handle, String fname)
    • XGBoosterSaveModel

      public static final int XGBoosterSaveModel(long handle, String fname)
    • XGBoosterLoadModelFromBuffer

      public static final int XGBoosterLoadModelFromBuffer(long handle, byte[] bytes)
    • XGBoosterGetModelRaw

      public static final int XGBoosterGetModelRaw(long handle, byte[][] out_bytes)
    • XGBoosterDumpModelEx

      public static final int XGBoosterDumpModelEx(long handle, String fmap, int with_stats, String format, String[][] out_strings)
    • XGBoosterDumpModelExWithFeatures

      public static final int XGBoosterDumpModelExWithFeatures(long handle, String[] feature_names, int with_stats, String format, String[][] out_strings)
    • XGBoosterGetAttrNames

      public static final int XGBoosterGetAttrNames(long handle, String[][] out_strings)
    • XGBoosterGetAttr

      public static final int XGBoosterGetAttr(long handle, String key, String[] out_string)
    • XGBoosterSetAttr

      public static final int XGBoosterSetAttr(long handle, String key, String value)
    • XGBoosterLoadRabitCheckpoint

      public static final int XGBoosterLoadRabitCheckpoint(long handle, int[] out_version)
    • XGBoosterSaveRabitCheckpoint

      public static final int XGBoosterSaveRabitCheckpoint(long handle)
    • RabitInit

      public static final int RabitInit(String[] args)
    • RabitFinalize

      public static final int RabitFinalize()
    • RabitTrackerPrint

      public static final int RabitTrackerPrint(String msg)
    • 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

      static final int RabitAllreduce(ByteBuffer sendrecvbuf, int count, int enum_dtype, int enum_op)