Class Dart

  • All Implemented Interfaces:
    GradBooster, java.io.Serializable

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

      • weightDrop

        private float[] weightDrop
    • Constructor Detail

      • Dart

        Dart()
    • Method Detail

      • loadModel

        public void loadModel​(ModelReader reader,
                              boolean with_pbuffer)
                       throws java.io.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:
        java.io.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