ml.dmlc.xgboost4j.java.spark.rapids

Class GpuColumnBatch

java.lang.Object

ml.dmlc.xgboost4j.java.spark.rapids.GpuColumnBatch

public class GpuColumnBatch
extends Object

Constructor Summary

Constructors

Constructor and Description
GpuColumnBatch(ai.rapids.cudf.Table table, org.apache.spark.sql.types.StructType schema)

Method Summary

Modifier and Type	Method and Description
long	getColumn(int index)
ai.rapids.cudf.ColumnVector	getColumnVector(int index)
int	getNumColumns()
long	getNumRows()
org.apache.spark.sql.types.StructType	getSchema()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

GpuColumnBatch

public GpuColumnBatch(ai.rapids.cudf.Table table,
                      org.apache.spark.sql.types.StructType schema)

Method Detail

getSchema

public org.apache.spark.sql.types.StructType getSchema()

getNumRows

public long getNumRows()

getNumColumns

public int getNumColumns()

getColumnVector

public ai.rapids.cudf.ColumnVector getColumnVector(int index)

getColumn

public long getColumn(int index)