Class GBTree

    • Field Detail

      • num_class

        protected int num_class
    • Method Detail

      • loadModel

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

        public float[] predict​(FVec feat,
                               int ntree_limit)
        Description copied from interface: GradBooster
        Generates predictions for given feature vector.
        Parameters:
        feat - feature vector
        ntree_limit - limit the number of trees used in prediction
        Returns:
        prediction result
      • predictSingle

        public float predictSingle​(FVec feat,
                                   int ntree_limit)
        Description copied from interface: GradBooster
        Generates a prediction for given feature vector.

        This method only works when the model outputs single value.

        Parameters:
        feat - feature vector
        ntree_limit - limit the number of trees used in prediction
        Returns:
        prediction result
      • predictLeaf

        public int[] predictLeaf​(FVec feat,
                                 int ntree_limit)
        Description copied from interface: GradBooster
        Predicts the leaf index of each tree. This is only valid in gbtree predictor.
        Parameters:
        feat - feature vector
        ntree_limit - limit the number of trees used in prediction
        Returns:
        predicted leaf indexes
      • setNumClass

        public void setNumClass​(int num_class)
        Specified by:
        setNumClass in interface GradBooster