Packages

class DMatrix extends AnyRef

DMatrix for xgboost.

Linear Supertypes
AnyRef, Any
Known Subclasses
QuantileDMatrix
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. DMatrix
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new DMatrix(columnBatch: ColumnBatch, missing: Float, nthread: Int)

    Create the normal DMatrix from column array interface

    Create the normal DMatrix from column array interface

    columnBatch

    the XGBoost ColumnBatch to provide the cuda array interface of feature columns

    missing

    missing value

    nthread

    threads number

    Exceptions thrown
  2. new DMatrix(handle: Long)

    used for DMatrix slice

    used for DMatrix slice

    Attributes
    protected[java]
  3. new DMatrix(matrix: BigDenseMatrix, missing: Float)

    create DMatrix from dense matrix

    create DMatrix from dense matrix

    matrix

    instance of BigDenseMatrix

    missing

    the specified value to represent the missing value

  4. new DMatrix(data: Array[Float], nrow: Int, ncol: Int, missing: Float)

    create DMatrix from dense matrix

    create DMatrix from dense matrix

    data

    data values

    nrow

    number of rows

    ncol

    number of columns

    missing

    the specified value to represent the missing value

  5. new DMatrix(matrix: BigDenseMatrix)

    create DMatrix from a BigDenseMatrix

    create DMatrix from a BigDenseMatrix

    matrix

    instance of BigDenseMatrix

    Exceptions thrown

    XGBoostError native error

  6. new DMatrix(headers: Array[Long], indices: Array[Int], data: Array[Float], st: SparseType, shapeParam: Int, missing: Float, nthread: Int)
  7. new DMatrix(headers: Array[Long], indices: Array[Int], data: Array[Float], st: SparseType, shapeParam: Int)

    Create DMatrix from Sparse matrix in CSR/CSC format.

    Create DMatrix from Sparse matrix in CSR/CSC format.

    headers

    The row index of the matrix.

    indices

    The indices of presenting entries.

    data

    The data content.

    st

    Type of sparsity.

    shapeParam

    when st is CSR, it specifies the column number, otherwise it is taken as row number

    Exceptions thrown
  8. new DMatrix(dataPath: String)

    Create DMatrix by loading libsvm file from dataPath

    Create DMatrix by loading libsvm file from dataPath

    dataPath

    The path to the data.

    Exceptions thrown
  9. new DMatrix(iter: Iterator[LabeledPoint], cacheInfo: String)
  10. new DMatrix(data: Array[Float], nrow: Int, ncol: Int)

    create DMatrix from dense matrix

    create DMatrix from dense matrix

    data

    data values

    nrow

    number of rows

    ncol

    number of columns

    Annotations
    @Deprecated
    Deprecated
    Exceptions thrown

    XGBoostError native error

  11. new DMatrix(headers: Array[Long], indices: Array[Int], data: Array[Float], st: SparseType)

    Create DMatrix from Sparse matrix in CSR/CSC format.

    Create DMatrix from Sparse matrix in CSR/CSC format.

    headers

    The row index of the matrix.

    indices

    The indices of presenting entries.

    data

    The data content.

    st

    Type of sparsity.

    Annotations
    @Deprecated
    Deprecated
    Exceptions thrown

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  6. def dispose(): Unit
  7. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  8. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  9. def finalize(): Unit
    Attributes
    protected[java]
    Definition Classes
    DMatrix → AnyRef
    Annotations
    @Override()
  10. def getBaseMargin(): Array[Float]

    Get base margin of the DMatrix.

  11. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  12. def getFeatureNames(): Array[String]

    Get feature names

    Get feature names

    returns

    an array of feature names to be returned

    Exceptions thrown
  13. def getFeatureTypes(): Array[String]

    Get feature types

    Get feature types

    returns

    an array of feature types to be returned

    Exceptions thrown
  14. def getGroup(): Array[Int]

    Get group sizes of DMatrix

    Get group sizes of DMatrix

    returns

    group size as array

    Exceptions thrown

    XGBoostError native error

  15. def getHandle(): Long

    Get the handle

  16. def getLabel(): Array[Float]

    get label values

    get label values

    returns

    label

    Exceptions thrown

    XGBoostError native error

  17. def getWeight(): Array[Float]

    get weight of the DMatrix

    get weight of the DMatrix

    returns

    weights

    Exceptions thrown

    XGBoostError native error

  18. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  19. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  20. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  21. def nonMissingNum(): Long

    Get the number of non-missing values of DMatrix.

    Get the number of non-missing values of DMatrix.

    returns

    The number of non-missing values

    Exceptions thrown

    XGBoostError native error

  22. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  23. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  24. def rowNum(): Long

    get the row number of DMatrix

    get the row number of DMatrix

    returns

    number of rows

    Exceptions thrown

    XGBoostError native error

  25. def saveBinary(filePath: String): Unit

    save DMatrix to filePath

  26. def setBaseMargin(baseMargin: Array[Array[Float]]): Unit

    Set base margin (initial prediction).

  27. def setBaseMargin(baseMargin: Array[Float]): Unit

    Set base margin (initial prediction).

    Set base margin (initial prediction).

    The margin must have the same number of elements as the number of rows in this matrix.

  28. def setBaseMargin(column: Column): Unit

    Set base margin of DMatrix from cuda array interface

    Set base margin of DMatrix from cuda array interface

    column

    the XGBoost Column to provide the cuda array interface of base margin column

    Exceptions thrown

    XGBoostError native error

  29. def setFeatureNames(values: Array[String]): Unit

    Set feature names

    Set feature names

    values

    feature names to be set

    Exceptions thrown
  30. def setFeatureTypes(values: Array[String]): Unit

    Set feature types

    Set feature types

    values

    feature types to be set

    Exceptions thrown
  31. def setGroup(group: Array[Int]): Unit

    Set group sizes of DMatrix (used for ranking)

    Set group sizes of DMatrix (used for ranking)

    group

    group size as array

    Exceptions thrown

    XGBoostError native error

  32. def setLabel(labels: Array[Float]): Unit

    set label of dmatrix

    set label of dmatrix

    labels

    labels

    Exceptions thrown

    XGBoostError native error

  33. def setLabel(column: Column): Unit

    Set label of DMatrix from cuda array interface

    Set label of DMatrix from cuda array interface

    column

    the XGBoost Column to provide the cuda array interface of label column

    Exceptions thrown

    XGBoostError native error

  34. def setWeight(weights: Array[Float]): Unit

    set weight of each instance

    set weight of each instance

    weights

    weights

    Exceptions thrown

    XGBoostError native error

  35. def setWeight(column: Column): Unit

    Set weight of DMatrix from cuda array interface

    Set weight of DMatrix from cuda array interface

    column

    the XGBoost Column to provide the cuda array interface of weight column

    Exceptions thrown

    XGBoostError native error

  36. def slice(rowIndex: Array[Int]): DMatrix

    Slice the DMatrix and return a new DMatrix that only contains rowIndex.

    Slice the DMatrix and return a new DMatrix that only contains rowIndex.

    rowIndex

    row index

    returns

    sliced new DMatrix

    Exceptions thrown

    XGBoostError native error

  37. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  38. def toString(): String
    Definition Classes
    AnyRef → Any
  39. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  40. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  41. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from AnyRef

Inherited from Any

Ungrouped