Class Dart

All Implemented Interfaces:
GradBooster, Serializable

public class Dart extends GBTree
Gradient boosted DART tree implementation.
See Also:
  • Field Details

    • weightDrop

      private float[] weightDrop
  • Constructor Details

    • Dart

      Dart()
  • Method Details

    • loadModel

      public void loadModel(ModelReader reader, boolean with_pbuffer) throws IOException
      Description copied from interface: GradBooster
      Loads model from stream.
      Specified by:
      loadModel in interface GradBooster
      Overrides:
      loadModel in class GBTree
      Parameters:
      reader - input stream
      with_pbuffer - whether the incoming data contains pbuffer
      Throws:
      IOException - If an I/O error occurs
    • pred

      float pred(FVec feat, int bst_group, int root_index, int ntree_limit)
      Overrides:
      pred in class GBTree