Interface RegTree

All Superinterfaces:
Serializable
All Known Implementing Classes:
RegTreeImpl
public interface RegTree extends Serializable
Regression tree.

  • Method Details

    • getLeafIndex

      int getLeafIndex(FVec feat)
      Retrieves nodes from root to leaf and returns leaf index.
      Parameters:
      feat - feature vector
      Returns:
      leaf index

    • getLeafPath

      void getLeafPath(FVec feat, StringBuilder sb)
      Retrieves nodes from root to leaf and returns path to leaf.
      Parameters:
      feat - feature vector
      sb - output param, will write path path to leaf into this buffer

    • getLeafValue

      float getLeafValue(FVec feat, int root_id)
      Retrieves nodes from root to leaf and returns leaf value.
      Parameters:
      feat - feature vector
      root_id - starting root index
      Returns:
      leaf value

    • getNodes

      RegTreeNode[] getNodes()
      Returns:
      Tree's nodes

    • getStats

      RegTreeNodeStat[] getStats()
      Returns:
      Tree's nodes stats