A B C G M N P S X

A

allocateGpuDevice() - Static method in class ml.dmlc.xgboost4j.java.XGBoostSparkJNI

B

buildUnsafeRows(long[]) - Static method in class ml.dmlc.xgboost4j.java.XGBoostSparkJNI

Build an array of fixed-length Spark UnsafeRow using the GPU.

C

createColumnarReader(FilePartition) - Method in interface ml.dmlc.xgboost4j.java.spark.rapids.PartitionReaderFactory

Returns a columnar partition reader to read data from the given FilePartition.

G

get() - Method in interface ml.dmlc.xgboost4j.java.spark.rapids.PartitionReader

Return the current record.

getColumn(int) - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getColumnVector(int) - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getColumnVectorInitHost(int) - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getGpuDevice() - Static method in class ml.dmlc.xgboost4j.java.XGBoostSparkJNI

getIntInColumn(int, int, int) - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getNumColumns() - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getNumRows() - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getRapidsType(DataType) - Static method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

getSchema() - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

GpuColumnBatch - Class in ml.dmlc.xgboost4j.java.spark.rapids

GpuColumnBatch(Table, StructType) - Constructor for class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

groupAndAggregateOnColumnsHost(int, int, int, List<Integer>, List<Float>) - Method in class ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

Group data by column "groupIndex", and do "count" aggregation on column "groupIndex", while do aggregation similar to "average" on column "weightIdx", but require values in a group are equal to each other, then merge the results with "groupInfo" and "weightInfo" separately.

M

ml.dmlc.xgboost4j.java - package ml.dmlc.xgboost4j.java

ml.dmlc.xgboost4j.java.spark.rapids - package ml.dmlc.xgboost4j.java.spark.rapids

N

next() - Method in interface ml.dmlc.xgboost4j.java.spark.rapids.PartitionReader

Proceed to next record, returns false if there is no more records.

P

PartitionReader<T> - Interface in ml.dmlc.xgboost4j.java.spark.rapids

A partition reader returned by { PartitionReaderFactory#createReader(InputPartition)} or { PartitionReaderFactory#createColumnarReader(InputPartition)}.

PartitionReaderFactory - Interface in ml.dmlc.xgboost4j.java.spark.rapids

A factory used to create PartitionReader instances.

S

supportColumnarReads(FilePartition) - Method in interface ml.dmlc.xgboost4j.java.spark.rapids.PartitionReaderFactory

Returns true if the given FilePartition should be read by Spark in a columnar way.

X

XGBoostSparkJNI - Class in ml.dmlc.xgboost4j.java

JNI functions for XGBoost4J-Spark

XGBoostSparkJNI() - Constructor for class ml.dmlc.xgboost4j.java.XGBoostSparkJNI

A B C G M N P S X