Package ml.dmlc.xgboost4j.java

Class XGBoostJNI

java.lang.Object
ml.dmlc.xgboost4j.java.XGBoostJNI
public 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
Author:
hzx

  • Constructor Details

    • XGBoostJNI

      public XGBoostJNI()

  • Method Details

    • XGBGetLastError

      public static final String XGBGetLastError()

    • XGDMatrixCreateFromFile

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

    • XGDMatrixCreateFromCSR

      public static final int XGDMatrixCreateFromCSR(long[] indptr, int[] indices, float[] data, int shapeParam, float missing, int nthread, long[] out)

    • XGDMatrixCreateFromCSC

      public static final int XGDMatrixCreateFromCSC(long[] colptr, int[] indices, float[] data, int shapeParam, float missing, int nthread, long[] out)

    • XGDMatrixCreateFromMat

      public static final int XGDMatrixCreateFromMat(float[] data, int nrow, int ncol, float missing, long[] out)

    • XGDMatrixCreateFromMatRef

      public static final int XGDMatrixCreateFromMatRef(long dataRef, 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)

    • XGDMatrixGetFloatInfo

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

    • XGDMatrixGetUIntInfo

      public static final int XGDMatrixGetUIntInfo(long handle, String filed, int[][] info)

    • XGDMatrixSetStrFeatureInfo

      public static final int XGDMatrixSetStrFeatureInfo(long handle, String field, String[] values)
      Set the feature information
      Parameters:
      handle - the DMatrix native address
      field - "feature_names" or "feature_types"
      values - an array of string
      Returns:
      0 when success, -1 when failure happens

    • XGDMatrixGetStrFeatureInfo

      public static final int XGDMatrixGetStrFeatureInfo(long handle, String field, long[] outLength, String[][] outValues)

    • XGDMatrixNumRow

      public static final int XGDMatrixNumRow(long handle, long[] row)

    • XGDMatrixNumNonMissing

      public static final int XGDMatrixNumNonMissing(long handle, long[] nonMissings)

    • 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)

    • XGBoosterTrainOneIter

      public static final int XGBoosterTrainOneIter(long handle, long dtrain, int iter, 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)

    • XGBoosterPredictFromDense

      public static final int XGBoosterPredictFromDense(long handle, float[] data, long nrow, long ncol, float missing, int iteration_begin, int iteration_end, int predict_type, float[] margin, 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)

    • XGBoosterSaveModelToBuffer

      public static final int XGBoosterSaveModelToBuffer(long handle, String format, 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)

    • XGBoosterGetNumFeature

      public static final int XGBoosterGetNumFeature(long handle, long[] feature)

    • XGBoosterGetNumBoostedRound

      public static final int XGBoosterGetNumBoostedRound(long handle, int[] rounds)

    • CommunicatorInit

      public static final int CommunicatorInit(String args)

    • CommunicatorFinalize

      public static final int CommunicatorFinalize()

    • CommunicatorPrint

      public static final int CommunicatorPrint(String msg)

    • CommunicatorGetRank

      public static final int CommunicatorGetRank(int[] out)

    • CommunicatorGetWorldSize

      public static final int CommunicatorGetWorldSize(int[] out)

    • TrackerCreate

      public static final int TrackerCreate(String host, int nWorkers, int port, int sortby, long timeout, long[] out)

    • TrackerRun

      public static final int TrackerRun(long handle)

    • TrackerWaitFor

      public static final int TrackerWaitFor(long handle, long timeout)

    • TrackerWorkerArgs

      public static final int TrackerWorkerArgs(long handle, long timeout, String[] out)

    • TrackerFree

      public static final int TrackerFree(long handle)

    • XGDMatrixSetInfoFromInterface

      public static final int XGDMatrixSetInfoFromInterface(long handle, String field, String json)

    • XGQuantileDMatrixCreateFromCallback

      public static final int XGQuantileDMatrixCreateFromCallback(Iterator<ColumnBatch> iter, long[] ref, String config, long[] out)

    • XGExtMemQuantileDMatrixCreateFromCallback

      public static final int XGExtMemQuantileDMatrixCreateFromCallback(Iterator<ColumnBatch> iter, long[] ref, String config, long[] out)

    • XGDMatrixCreateFromArrayInterfaceColumns

      public static final int XGDMatrixCreateFromArrayInterfaceColumns(String featureJson, float missing, int nthread, long[] out)

    • XGBoosterSetStrFeatureInfo

      public static final int XGBoosterSetStrFeatureInfo(long handle, String field, String[] features)

    • XGBoosterGetStrFeatureInfo

      public static final int XGBoosterGetStrFeatureInfo(long handle, String field, String[] out)

    • XGDMatrixGetQuantileCut

      public static final int XGDMatrixGetQuantileCut(long handle, long[][] outIndptr, float[][] outValues)

    • XGBSetGlobalConfig

      public static final int XGBSetGlobalConfig(String config)

    • XGBGetGlobalConfig

      public static final int XGBGetGlobalConfig(String[] out)

    • CudaSetDevice

      public static final int CudaSetDevice(int deviceId)