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

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

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

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

This is used to group the CUDF Dataset by column 'groupIndex', and merge all the rows into one row in each group based on column 'oneIndex'.

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

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