Class XGBoostTrainer.DMatrixTuple<T extends Output<T>>

java.lang.Object
org.tribuo.common.xgboost.XGBoostTrainer.DMatrixTuple<T>
Type Parameters:
T - The output type.
Enclosing class:
XGBoostTrainer<T extends Output<T>>

protected static class XGBoostTrainer.DMatrixTuple<T extends Output<T>> extends Object
Tuple of a DMatrix, the number of valid features in each example, and the examples themselves.

One day it'll be a record.

  • Field Details

    • data

      public final ml.dmlc.xgboost4j.java.DMatrix data
      The data matrix.
    • numValidFeatures

      public final int[] numValidFeatures
      The number of valid features in each example.
    • examples

      public final Example<T extends Output<T>>[] examples
      The examples.
  • Constructor Details

    • DMatrixTuple

      protected DMatrixTuple(ml.dmlc.xgboost4j.java.DMatrix data, int[] numValidFeatures, Example<T>[] examples)
      Constructs a tuple containing the data and some Tribuo metadata.
      Parameters:
      data - The data matrix.
      numValidFeatures - The number of valid features in each example.
      examples - The examples.